diff --git a/.github/scripts/run-multi-zh_hans-zipformer.sh b/.github/scripts/run-multi-zh_hans-zipformer.sh
index 63b84b059..1392b8429 100755
--- a/.github/scripts/run-multi-zh_hans-zipformer.sh
+++ b/.github/scripts/run-multi-zh_hans-zipformer.sh
@@ -33,7 +33,7 @@ for sym in 1 2 3; do
 
   ./zipformer/pretrained.py \
     --checkpoint $repo/exp/epoch-99.pt \
-    --lang-dir $repo/data/lang_bpe_2000 \
+    --tokens $repo/data/lang_bpe_2000/tokens.txt \
     --decoding-method greedy_search \
     --max-sym-per-frame $sym \
   $repo/test_wavs/DEV_T0000000000.wav \
@@ -48,7 +48,7 @@ for method in modified_beam_search beam_search fast_beam_search; do
     --decoding-method $method \
     --beam-size 4 \
     --checkpoint $repo/exp/epoch-99.pt \
-    --lang-dir $repo/data/lang_bpe_2000 \
+    --tokens $repo/data/lang_bpe_2000/tokens.txt \
   $repo/test_wavs/DEV_T0000000000.wav \
   $repo/test_wavs/DEV_T0000000001.wav \
   $repo/test_wavs/DEV_T0000000002.wav