diff --git a/distributed_train.sh b/distributed_train.sh
index a1c68c0a..88406997 100755
--- a/distributed_train.sh
+++ b/distributed_train.sh
@@ -1,5 +1,5 @@
 #!/bin/bash
 NUM_PROC=$1
 shift
-python -m torch.distributed.launch --nproc_per_node=$NUM_PROC dtrain.py "$@"
+python -m torch.distributed.launch --nproc_per_node=$NUM_PROC train.py "$@"