diff --git a/tests/test_functional.py b/tests/test_functional.py
index 5b7f83bc3..340278912 100644
--- a/tests/test_functional.py
+++ b/tests/test_functional.py
@@ -613,9 +613,9 @@ def test_vector_quant(dim1, dim2, dim3):
 
 @pytest.mark.parametrize("dim1, dim2, dim3, dims, dtype, orderA, orderOut, transpose",values,ids=names)
 def test_nvidia_transform(dim1, dim2, dim3, dims, dtype, orderA, orderOut, transpose):
-    if dims == 3 and out_order != "col32":
+    if dims == 3 and orderOut != "col32":
         return
-    if dtype == torch.int32 and out_order != "col32":
+    if dtype == torch.int32 and orderOut != "col32":
         return
     try:
         func = F.get_transform_func(dtype, orderA, orderOut, transpose)