diff --git a/main_heuristics.py b/main_heuristics.py
index 2853150..dd56e27 100644
--- a/main_heuristics.py
+++ b/main_heuristics.py
@@ -98,7 +98,8 @@ def choose_gmods(features):
 #         metrics = compute_metrics(chosen_indices,
 #                                   testing_dataset['labels'],
 #                                   testing_dataset['timings'],
-#                                   testing_dataset['cells'])
+#                                   testing_dataset['cells'],
+#                                   heuristic)
 #         if len(sum_metrics) == 0:
 #             sum_metrics = metrics
 #         else:
diff --git a/train_models.py b/train_models.py
index 5aeede2..bde4209 100644
--- a/train_models.py
+++ b/train_models.py
@@ -80,7 +80,7 @@ def train_reinforcement_model(ml_model, method='Normal'):
     model = current_model()
     first_polys = train_dataset['projections'][0][0][0]
     first_features = get_vars_features(first_polys)
-    first_labels = [random.random() for _ in range([len(first_features)])]
+    first_labels = [random.random() for _ in range(len(first_features))]
     model.fit(first_features, first_labels)
     training_features, training_labels = [], []
     for i in range(30):