diff --git a/scripts/hpc/run_slurm.sh b/scripts/hpc/run_slurm.sh
index 849e7ae..0d14424 100644
--- a/scripts/hpc/run_slurm.sh
+++ b/scripts/hpc/run_slurm.sh
@@ -8,6 +8,6 @@ fi
 
 source $COMMON_SLURM_PATH
 
-PYTHON_COMMAND="PYTHONPATH='.' python3 -u ${DEEPFRI_PATH}/FRIdata/fridata.py input_generation -t $1 -d AFDB -c subset --overwrite --version 1_test_dask -i ${IDS_PATH} --input-path ${AFDB_PATH} -e ${EMBEDDER_TYPE} --slurm --verbose"
+PYTHON_COMMAND="PYTHONPATH='.' python3 -u ${DEEPFRI_PATH}/FRIdata/fridata.py generate_data -t $1 -d AFDB -c subset --overwrite --version 1_test_dask -i ${IDS_PATH} --input-path ${AFDB_PATH} -e ${EMBEDDER_TYPE} --slurm --verbose"
 
 start_computation "$PYTHON_COMMAND"
\ No newline at end of file