diff --git a/machine_learning/helper.py b/machine_learning/helper.py
index 0b7574c..177f608 100644
--- a/machine_learning/helper.py
+++ b/machine_learning/helper.py
@@ -182,7 +182,7 @@ def run_all_regression_models(
 ) -> pd.DataFrame:
     metrics = ["r2", "neg_mean_absolute_error", "neg_root_mean_squared_error"]
     test_metrics = ["test_" + metric for metric in metrics]
-    scores = pd.DataFrame(columns=["method", "metric", "max", "nanmedian"])
+    scores = pd.DataFrame(columns=["method", "test_metric", "max", "nanmedian"])
 
     # Validate models
     dummy_regr = DummyRegressor(strategy="mean")
@@ -427,7 +427,7 @@ def run_all_classification_models(
     metrics = ["accuracy", "average_precision", "recall", "f1"]
     test_metrics = ["test_" + metric for metric in metrics]
 
-    scores = pd.DataFrame(columns=["method", "metric", "max", "mean"])
+    scores = pd.DataFrame(columns=["method", "test_metric", "max", "mean"])
 
     dummy_class = DummyClassifier(strategy="most_frequent")