diff --git a/freqtrade/freqai/data_drawer.py b/freqtrade/freqai/data_drawer.py
index 50cc70fe0..4df52fdc4 100644
--- a/freqtrade/freqai/data_drawer.py
+++ b/freqtrade/freqai/data_drawer.py
@@ -285,7 +285,7 @@ class FreqaiDataDrawer:
 
         new_pred["date_pred"] = dataframe["date"]
         # set everything to nan except date_pred
-        columns_to_nan = new_pred.columns.difference(['date_pred'])
+        columns_to_nan = new_pred.columns.difference(['date_pred', 'date'])
         new_pred[columns_to_nan] = new_pred[columns_to_nan].astype(
             float).values * np.nan