diff --git a/egs/aishell/paraformerbert/run.sh b/egs/aishell/paraformerbert/run.sh
index e65ad2c71..a4ad195aa 100755
--- a/egs/aishell/paraformerbert/run.sh
+++ b/egs/aishell/paraformerbert/run.sh
@@ -115,7 +115,8 @@ if [ ${stage} -le 3 ] && [ ${stop_stage} -ge 3 ]; then
         echo "extract embeddings..."
         local/extract_embeds.sh \
             --bert_model_name ${bert_model_name} \
-            --raw_dataset_path ${feats_dir}
+            --raw_dataset_path ${feats_dir} \
+            --nj $nj
     fi
     mkdir -p ${exp_dir}/exp/${model_dir}
     mkdir -p ${exp_dir}/exp/${model_dir}/log
diff --git a/egs/aishell2/paraformerbert/run.sh b/egs/aishell2/paraformerbert/run.sh
index eb2252bfc..26c8063ad 100755
--- a/egs/aishell2/paraformerbert/run.sh
+++ b/egs/aishell2/paraformerbert/run.sh
@@ -116,7 +116,8 @@ if [ ${stage} -le 3 ] && [ ${stop_stage} -ge 3 ]; then
         echo "extract embeddings..."
         local/extract_embeds.sh \
             --bert_model_name ${bert_model_name} \
-            --raw_dataset_path ${feats_dir}
+            --raw_dataset_path ${feats_dir} \
+            --nj $nj
     fi
     mkdir -p ${exp_dir}/exp/${model_dir}
     mkdir -p ${exp_dir}/exp/${model_dir}/log