diff --git a/PaddleNLP/language_model/.run_ce.sh b/PaddleNLP/language_model/.run_ce.sh
index 96c7e71defd5c14675ce3303bc652b43dd1cd484..4eda4317db14ed27683b2c6293b2d1620a10e821 100644
--- a/PaddleNLP/language_model/.run_ce.sh
+++ b/PaddleNLP/language_model/.run_ce.sh
@@ -2,14 +2,14 @@ export CUDA_VISIBLE_DEVICES=0
 
 python  train.py \
     --data_path data/simple-examples/data/ \
-    --model_type test \
+    --model_type small \
     --use_gpu True \
     --rnn_model static \
     --enable_ce | python _ce.py
 
 python  train.py \
     --data_path data/simple-examples/data/ \
-    --model_type test \
+    --model_type small \
     --use_gpu True \
     --rnn_model padding \
     --enable_ce | python _ce.py