diff --git a/dags/config.py b/dags/config.py
index 07fbc05..7bbe5e0 100644
--- a/dags/config.py
+++ b/dags/config.py
@@ -6,9 +6,9 @@
 GENERATED_DATA_FOLDER = os.path.join(DATA_FOLDER, "generated_data")
 PREDICTIONS_FOLDER = os.path.join(DATA_FOLDER, 'predictions')
 
-TRAIN_DATA_PATH = os.path.join(DATA_FOLDER, 'la-haute-borne-data-2013-2016.csv')
+TRAIN_DATA_PATH = os.path.join(DATA_FOLDER, 'la-haute-borne-data-2017-2020.csv')
 # TODO change, should point to generator
-GENERATED_DATA_PATH = os.path.join(DATA_FOLDER, 'la-haute-borne-data-2013-2016.csv')
+GENERATED_DATA_PATH = os.path.join(DATA_FOLDER, 'la-haute-borne-data-2017-2020.csv')
 
 FEATURES_PATH = os.path.join(DATA_FOLDER, 'prepared_features.parquet')