diff --git a/test.py b/test.py
index 60154a69cf..894fa08ebb 100644
--- a/test.py
+++ b/test.py
@@ -59,7 +59,7 @@ def test(data,
         imgsz = check_img_size(imgsz, s=gs)  # check img_size
         
         if trace:
-            model = TracedModel(model, device, opt.img_size)
+            model = TracedModel(model, device, imgsz)
 
     # Half
     half = device.type != 'cpu' and half_precision  # half precision only supported on CUDA