diff --git a/src/test.py b/src/test.py
index 1d08044d..dd62999d 100755
--- a/src/test.py
+++ b/src/test.py
@@ -33,8 +33,8 @@ def __getitem__(self, index):
     img_path = os.path.join(self.img_dir, img_info['file_name'])
     image = cv2.imread(img_path)
     images, meta = {}, {}
-    for scale in opt.test_scales:
-      if opt.task == 'ddd':
+    for scale in self.opt.test_scales:
+      if self.opt.task == 'ddd':
         images[scale], meta[scale] = self.pre_process_func(
           image, scale, img_info['calib'])
       else:
@@ -123,4 +123,4 @@ def test(opt):
   if opt.not_prefetch_test:
     test(opt)
   else:
-    prefetch_test(opt)
\ No newline at end of file
+    prefetch_test(opt)