diff --git a/laborious/activities/mlflow.py b/laborious/activities/mlflow.py
index ba6a346..3ac5d95 100644
--- a/laborious/activities/mlflow.py
+++ b/laborious/activities/mlflow.py
@@ -529,6 +529,7 @@ class MLFlow(SientiaMonitoring):
 
             # Keep heavy model fitting outside MLflow run timing.
             prediction_data = wrapper.retrain(data)
+            prediction_data.rename(columns={target: 'prediction'}, inplace=True)
 
             # Merge prediction data with retrain data
             evaluation_data = pd.merge(