diff --git a/egs/gigaspeech/KWS/run.sh b/egs/gigaspeech/KWS/run.sh
index ea04c7c9b4..42e864efee 100755
--- a/egs/gigaspeech/KWS/run.sh
+++ b/egs/gigaspeech/KWS/run.sh
@@ -123,6 +123,7 @@ if [ $stage -le 3 ] && [ $stop_stage -ge 3 ]; then
     --exp-dir zipformer/exp_finetune \
     --bpe-model data/lang_bpe_500/bpe.model \
     --use-fp16 1 \
+    --use-mux 1 \
     --decoder-dim 320 \
     --joiner-dim 320 \
     --num-encoder-layers 1,1,1,1,1,1 \
diff --git a/egs/wenetspeech/KWS/run.sh b/egs/wenetspeech/KWS/run.sh
index 2bdd6a5f34..f702e08175 100755
--- a/egs/wenetspeech/KWS/run.sh
+++ b/egs/wenetspeech/KWS/run.sh
@@ -126,6 +126,7 @@ if [ $stage -le 3 ] && [ $stop_stage -ge 3 ]; then
     --lang-dir ./data/lang_partial_tone \
     --pinyin-type partial_with_tone \
     --use-fp16 1 \
+    --use-mux 1 \
     --decoder-dim 320 \
     --joiner-dim 320 \
     --num-encoder-layers 1,1,1,1,1,1 \