diff --git a/test/test_prims.py b/test/test_prims.py
index 7b616f2e242fb..2d78acb68fc94 100644
--- a/test/test_prims.py
+++ b/test/test_prims.py
@@ -538,10 +538,10 @@ def func4(a):
             return torch.reshape(a, tuple(reversed(a.shape)))
 
         def func5(a):
-            return torch.ops.aten.view(a, tuple(reversed(a.shape)))
+            return torch.ops.aten.view.default(a, tuple(reversed(a.shape)))
 
         def func6(a):
-            return torch.ops.aten.view.default(a, tuple(reversed(a.shape)))
+            return torch.ops.aten._unsafe_view.default(a, tuple(reversed(a.shape)))
 
         for func in (func1, func2, func3, func4, func5, func6):
             with TorchRefsNvfuserCapabilityMode():