diff --git a/HPC_scripts/DataPreprocess_to_tf.sh b/HPC_scripts/DataPreprocess_to_tf.sh
index e984ef0ca16aada67aa47f66714f2400f73e967b..667682d28d38bd720de86f40f45384c22a4a0baa 100755
--- a/HPC_scripts/DataPreprocess_to_tf.sh
+++ b/HPC_scripts/DataPreprocess_to_tf.sh
@@ -19,4 +19,4 @@ module load h5py/2.9.0-Python-3.6.8
 module load mpi4py/3.0.1-Python-3.6.8
 
 
-srun python ../video_prediction/datasets/era5_dataset_v2.py  /p/scratch/deepacf/bing/processData_size_64_64_3_3t_norm  ../data/era5_64_64_3_3t_norm
+srun python ../video_prediction/datasets/era5_dataset_v2.py  /p/scratch/deepacf/bing/processData_size_64_64_3_3t_norm/splits  ../data/era5_size_64_64_3_3t_norm