diff --git a/PaddleNLP/pretrain_language_models/BERT/train.sh b/PaddleNLP/pretrain_language_models/BERT/train.sh
index 4dd00e0d547845d54a334843551f7c22b471ab13..2ed44e4c7291f502c70bfad4422f7ad999439c58 100755
--- a/PaddleNLP/pretrain_language_models/BERT/train.sh
+++ b/PaddleNLP/pretrain_language_models/BERT/train.sh
@@ -32,7 +32,7 @@ VALIDATION_DATA_DIR=data/validation
 CONFIG_PATH=data/demo_config/bert_config.json
 VOCAB_PATH=data/demo_config/vocab.txt
 # Change your train arguments:
-python3 -u ./train.py ${is_distributed}\
+python -u ./train.py ${is_distributed}\
         --use_cuda true\
         --weight_sharing true\
         --batch_size ${BATCH_SIZE} \