diff --git a/results/result_generation.py b/results/result_generation.py
index c84383a301491cfd360cf94fea764f4687f085c4..caa42642976785d82f0313733b324c9cd63b1166 100644
--- a/results/result_generation.py
+++ b/results/result_generation.py
@@ -79,7 +79,7 @@ def get_scale_factors(group):
         "parallel_fed_time",
     ]
     scale_factors = []
-    num_runs = len(next(iter(group))[1])
+    num_runs = len(next(iter(group.values())))
     max_runtime = [0] * num_runs
     elapsed_time = [0] * num_runs
     parallel_time = [0] * num_runs