diff --git a/.github/workflows/unitest.yml b/.github/workflows/unitest.yml
index ee691d1..464d799 100644
--- a/.github/workflows/unitest.yml
+++ b/.github/workflows/unitest.yml
@@ -32,5 +32,5 @@ jobs:
         touch logs/mi-test/preprocess.log
         python trainset_preprocess_pipeline_print.py logs/mute/0_gt_wavs 48000 8 logs/mi-test True
         touch logs/mi-test/extract_f0_feature.log
-        python extract_f0_print.py logs/mi-test $(nproc) pm
+        python extract_f0_print.py logs/mi-test $(nproc) pm v1
         python extract_feature_print.py cpu 1 0 0 logs/mi-test
diff --git a/extract_feature_print.py b/extract_feature_print.py
index fa46afb..abf4d25 100644
--- a/extract_feature_print.py
+++ b/extract_feature_print.py
@@ -5,11 +5,12 @@ n_part = int(sys.argv[2])
 i_part = int(sys.argv[3])
 if len(sys.argv) == 5:
     exp_dir = sys.argv[4]
+    version = sys.argv[5]
 else:
     i_gpu = sys.argv[4]
     exp_dir = sys.argv[5]
     os.environ["CUDA_VISIBLE_DEVICES"] = str(i_gpu)
-version = sys.argv[6]
+    version = sys.argv[6]
 import torch
 import torch.nn.functional as F
 import soundfile as sf