diff --git a/src/test.py b/src/test.py
index bcca0e3..486d000 100644
--- a/src/test.py
+++ b/src/test.py
@@ -33,7 +33,7 @@ def main_worker(args, use_gpu=True):
     # prepare dataset
     image_paths = []
     for ext in ['.jpg', '.png']: 
-        image_paths.extend(glob(os.path.join(args.dir_image, '*'+ext)))
+        image_paths.extend(glob(os.path.join(args.dir_test, '*'+ext)))
     image_paths.sort()
     mask_paths = sorted(glob(os.path.join(args.dir_mask,args.mask_type,'*.png')))
     os.makedirs(args.outputs, exist_ok=True)