diff --git a/dsv4/kernels/attention/fmha_6warp_tma.cuh b/dsv4/kernels/attention/fmha_6warp_tma.cuh
index 2b298c36..c0f97519 100644
--- a/dsv4/kernels/attention/fmha_6warp_tma.cuh
+++ b/dsv4/kernels/attention/fmha_6warp_tma.cuh
@@ -160,7 +160,7 @@ fmha_6warp_tma_kernel(FmhaMultiRowTmaParams params) {
     // ==================================================================
     if (tid == 0) {
         uint32_t mbar_addr = (uint32_t)__cvta_generic_to_shared(sMbar);
-        tma_mbar_init(mbar_addr);
+        tma_mbarrier_init(mbar_addr);
     }
 
     // TMEM alloc