diff --git a/Data/generate_vocab.py b/Data/generate_vocab.py
index 579c5f7..bb883aa 100644
--- a/Data/generate_vocab.py
+++ b/Data/generate_vocab.py
@@ -58,7 +58,7 @@ def write_vocab_file(filepath, vocab):
     parser = argparse.ArgumentParser()
     parser.add_argument("-d", "--dir", help="The dir should contain several sub_dirs which contain text files belong to different categories", type=str, required=True) 
     parser.add_argument("-f", "--file", help="The file is the final generated feature vocabulary file.", type=str, required=True)
-    parser.add_argument("s", "--size", help="The target vocabulary size.", type=int, default=10000)
+    parser.add_argument("-s", "--size", help="The target vocabulary size.", type=int, default=10000)
     args = parser.parse_args()
 
-    gen_vocab(dataset_dir=args.dir, vocab_file=args.file, vocab_size=args.size)
\ No newline at end of file
+    gen_vocab(dataset_dir=args.dir, vocab_file=args.file, vocab_size=args.size)
diff --git a/Model/pretrainDexBERT.sh b/Model/pretrainDexBERT.sh
index 81a5925..e9f528f 100644
--- a/Model/pretrainDexBERT.sh
+++ b/Model/pretrainDexBERT.sh
@@ -4,7 +4,7 @@ export SAVE_DIR=../save_dir/DexBERT
 export LOG_DIR=../log_dir/DexBERT
 
 python pretrainDexBERT.py \
-    --train_cfg confi/DexBERT/pretrain.json \
+    --train_cfg config/DexBERT/pretrain.json \
     --model_cfg config/DexBERT/bert_base.json \
     --data_file $DATA_FILE \
     --vocab $VOCAB_FILE \