diff --git a/predict.py b/predict.py
index 59ebaa2..c30880d 100644
--- a/predict.py
+++ b/predict.py
@@ -39,7 +39,7 @@ def predict(model_path: Union[str, IO],
 
     res = model.validation_step(graph, batch_idx=0)[0]
     end = time.time()
-    res['run_time'] = end - start,
+    res['run_time'] = end - start
     pprint(res, sort_dicts=False)