diff --git a/fluid/language_model/.run_ce.sh b/fluid/language_model/.run_ce.sh
index c0cb8f5473a100ebf0e5c0697f9ab5e65e97560d..5ee2d8aa0582b2b8504f9ba645b6252aa75f23bf 100644
--- a/fluid/language_model/.run_ce.sh
+++ b/fluid/language_model/.run_ce.sh
@@ -6,9 +6,9 @@ export OMP_NUM_THREADS=1
 cudaid=${language_model:=0} # use 0-th card as default
 export CUDA_VISIBLE_DEVICES=$cudaid
 
-FLAGS_benchmark=true  python train.py | python _ce.py
+FLAGS_benchmark=true  python train.py --enable_ce | python _ce.py
 
 cudaid=${language_model_m:=0,1,2,3} # use 0,1,2,3 card as default
 export CUDA_VISIBLE_DEVICES=$cudaid
 
-FLAGS_benchmark=true  python train.py | python _ce.py
+FLAGS_benchmark=true  python train.py --enable_ce | python _ce.py