diff --git a/tests/unit/test_d1_qk512.py b/tests/unit/test_d1_qk512.py
index 581187ee..2fa4f249 100644
--- a/tests/unit/test_d1_qk512.py
+++ b/tests/unit/test_d1_qk512.py
@@ -5,7 +5,7 @@ import cutlass.cute as cute
 import cutlass.torch as ct
 import cuda.bindings.driver as cuda
 from cutlass import BFloat16, Float32
-from cutlass.cute.nvgpu import tcgen05
+from cutlass.cute.nvgpu import cpasync, tcgen05
 from cutlass.utils import LayoutEnum
 import cutlass.utils as utils
 import cutlass.pipeline as pipeline