diff --git a/tests/unit/test_nvfp4_cutedsl_compile.py b/tests/unit/test_nvfp4_cutedsl_compile.py
index 430a0457..e1ab9870 100644
--- a/tests/unit/test_nvfp4_cutedsl_compile.py
+++ b/tests/unit/test_nvfp4_cutedsl_compile.py
@@ -54,8 +54,10 @@ def test_nvfp4_cutedsl_compilation():
     sfb_tensor = cutlass_torch.from_dlpack(scale_b)
     sfb_tensor = sfb_tensor.mark_layout_dynamic(leading_dim=cutlass_torch.get_leading_dim(scale_b))
 
-    c_tensor = cutlass_torch.as_tensor(
+    c_tensor = cutlass_torch.from_dlpack(
         torch.empty(M, N, dtype=torch.bfloat16, device=device))
+    c_tensor = c_tensor.mark_layout_dynamic(leading_dim=cutlass_torch.get_leading_dim(
+        torch.empty(M, N, dtype=torch.bfloat16, device=device)))
 
     print("CuTe tensors created OK")