diff --git a/mnist_ddp_deepspeed.py b/mnist_deepspeed.py
similarity index 100%
rename from mnist_ddp_deepspeed.py
rename to mnist_deepspeed.py
diff --git a/run-deepspeed-gpu4.sh b/run-deepspeed-gpu4.sh
index 68c3959..24eb7bf 100755
--- a/run-deepspeed-gpu4.sh
+++ b/run-deepspeed-gpu4.sh
@@ -10,5 +10,5 @@
 module purge
 module load pytorch
 
-srun singularity_wrapper exec deepspeed mnist_ddp_deepspeed.py --epochs=100 \
+srun singularity_wrapper exec deepspeed mnist_deepspeed.py --epochs=100 \
      --deepspeed --deepspeed_config ds_config.json
diff --git a/run-deepspeed-gpu8.sh b/run-deepspeed-gpu8.sh
index 9c4d71d..5b80279 100755
--- a/run-deepspeed-gpu8.sh
+++ b/run-deepspeed-gpu8.sh
@@ -11,5 +11,5 @@
 module purge
 module load pytorch
 
-srun python3 mnist_ddp_deepspeed.py --epochs=100 \
+srun python3 mnist_deepspeed.py --epochs=100 \
      --deepspeed --deepspeed_config ds_config.json