diff --git a/tools/benchmarks/classification/svm_voc07/extract.py b/tools/benchmarks/classification/svm_voc07/extract.py
index f69b177f..0a0da17a 100644
--- a/tools/benchmarks/classification/svm_voc07/extract.py
+++ b/tools/benchmarks/classification/svm_voc07/extract.py
@@ -149,13 +149,13 @@ def main():
     # run
     outputs = extractor.extract(model, data_loader, distributed=distributed)
     rank, _ = get_dist_info()
-    mmcv.mkdir_or_exist(f'{args.work_dir}/features/')
+    mmcv.mkdir_or_exist(f'{cfg.work_dir}/features/')
     if rank == 0:
         for key, val in outputs.items():
             split_num = len(dataset_cfg.split_name)
             split_at = dataset_cfg.split_at
             for ss in range(split_num):
-                output_file = f'{args.work_dir}/features/' \
+                output_file = f'{cfg.work_dir}/features/' \
                               f'{dataset_cfg.split_name[ss]}_{key}.npy'
                 if ss == 0:
                     np.save(output_file, val[:split_at[0]])