diff --git a/tensorflow/jureca_job.sh b/tensorflow/jureca_job.sh
index e818bc030359232fb55d097a776828b4e56aa61d..864c5ec15e3110837fb63e2115f367cecb69ab84 100755
--- a/tensorflow/jureca_job.sh
+++ b/tensorflow/jureca_job.sh
@@ -21,4 +21,4 @@ module load Horovod/0.20.3-Python-3.8.5
 export CUDA_VISIBLE_DEVICES=0,1,2,3
 
 # Run the program
-srun python -u mnist.py
+srun python -u keras_mnist.py
diff --git a/tensorflow/jusuf_job.sh b/tensorflow/jusuf_job.sh
index 24f3c83acc514255de6f3fd0452cfda8ab553da2..1c957b66c30db39398bf41544fae2833d190e1d8 100755
--- a/tensorflow/jusuf_job.sh
+++ b/tensorflow/jusuf_job.sh
@@ -21,4 +21,4 @@ module load Horovod/0.20.3-Python-3.8.5
 export CUDA_VISIBLE_DEVICES=0
 
 # Run the program
-srun python -u mnist.py
+srun python -u keras_mnist.py
diff --git a/tensorflow/juwels_booster_job.sh b/tensorflow/juwels_booster_job.sh
index df9cdef250c5c5aed3006f53ba2e32fad6a46db3..f8b2b22dd4c80ed4fc9a4daa3b5fed96d61ba3c3 100755
--- a/tensorflow/juwels_booster_job.sh
+++ b/tensorflow/juwels_booster_job.sh
@@ -21,4 +21,4 @@ module load Horovod/0.20.3-Python-3.8.5
 export CUDA_VISIBLE_DEVICES=0,1,2,3
 
 # Run the program
-srun python -u mnist.py
+srun python -u keras_mnist.py
diff --git a/tensorflow/juwels_job.sh b/tensorflow/juwels_job.sh
index 55831d0785c0bba72bd68a2339a18997cc45dd69..7cae2f5b38e4cd8ce8c914ff17b0dbdea35b1922 100755
--- a/tensorflow/juwels_job.sh
+++ b/tensorflow/juwels_job.sh
@@ -21,4 +21,4 @@ module load Horovod/0.20.3-Python-3.8.5
 export CUDA_VISIBLE_DEVICES=0,1,2,3
 
 # Run the program
-srun python -u mnist.py
+srun python -u keras_mnist.py