diff --git a/video_prediction_tools/main_scripts/main_preprocess_data_step2.py b/video_prediction_tools/main_scripts/main_preprocess_data_step2.py
index d5a57b8d173c65e5b8c7bda459053bc21eaf07ab..51894163489456ab72d330dc40e4af6936be6e36 100644
--- a/video_prediction_tools/main_scripts/main_preprocess_data_step2.py
+++ b/video_prediction_tools/main_scripts/main_preprocess_data_step2.py
@@ -18,6 +18,7 @@ import warnings
 def main():
     parser = argparse.ArgumentParser()
     parser.add_argument("-input_dir", type=str)
+    parser.add_argument("-dest_dir", type=str)
     parser.add_argument("-sequence_length", type=int, default=20)
     parser.add_argument("-sequences_per_file", type=int, default=20)
     args = parser.parse_args()
@@ -90,6 +91,7 @@ def main():
             if year_rank in real_years_months:
                 # Initilial instance
                 ins2 = ERA5Pkl2Tfrecords(input_dir=args.input_dir,
+                                         dest_dir=args.dest_dir,
                                          sequence_length = args.sequence_length,
                                          sequences_per_file=args.sequences_per_file)
                 # create the tfrecords-files