diff --git a/compare_graph_maker.py b/compare_graph_maker.py
index 7d6c02d..3b4d855 100644
--- a/compare_graph_maker.py
+++ b/compare_graph_maker.py
@@ -18,7 +18,7 @@ for dataset_name, folder in datasets.items():
                 for line in f:
                     if line.strip().startswith("weighted avg"):
                         parts = line.split()
-                        f1_score = float(parts[3])  # precision recall f1-score support
+                        f1_score = float(parts[4])  # precision recall f1-score support
                         results[dataset_name][model_name] = f1_score
 
 # Plot