diff --git a/video_prediction_savp/video_prediction/datasets/era5_dataset_v2.py b/video_prediction_savp/video_prediction/datasets/era5_dataset_v2.py
index 63cff248937b9bfd6ff6793d16d665ed94e6dcf5..2fb693e4066570a93889850c571654a46fb81e4e 100644
--- a/video_prediction_savp/video_prediction/datasets/era5_dataset_v2.py
+++ b/video_prediction_savp/video_prediction/datasets/era5_dataset_v2.py
@@ -371,12 +371,12 @@ def main():
     ############################################################
     partition = {
             "train":{
-                "2222":[1,2,3,5,6,7,8,9,10,11,12], # Issue due to month 04, it is missing
-           #     "2010":[1,2,3,4,5,6,7,8,9,10,11,12],
+           #     "2222":[1,2,3,5,6,7,8,9,10,11,12], # Issue due to month 04, it is missing
+                "2010":[1,2,3,4,5,6,7,8,9,10,11,12],
            #     "2012":[1,2,3,4,5,6,7,8,9,10,11,12],
-           #     "2013":[1,2,3,4,5,6,7,8,9,10,11,12],
-           #     "2015":[1,2,3,4,5,6,7,8,9,10,11,12],
-           #     "2019":[1,2,3,4,5,6,7,8,9,10,11,12]
+                "2013":[1,2,3,4,5,6,7,8,9,10,11,12],
+                "2015":[1,2,3,4,5,6,7,8,9,10,11,12],
+                "2019":[1,2,3,4,5,6,7,8,9,10,11,12]
                  },
             "val":
                 {"2017":[1,2,3,4,5,6,7,8,9,10,11,12]