diff --git a/dsv4/kernels/attention/fmha_smem_acc.py b/dsv4/kernels/attention/fmha_smem_acc.py
index 8e88b4bd..e44a3857 100644
--- a/dsv4/kernels/attention/fmha_smem_acc.py
+++ b/dsv4/kernels/attention/fmha_smem_acc.py
@@ -628,6 +628,8 @@ class FmhaKernel:
                     cute.group_modes(sC, 0, 2),
                     cute.group_modes(tCgC_epi, 0, 2),
                 )
+                # Slice off MMA tile coordinates (same as epilogue_tma_store)
+                bSG_gC = bSG_gC[(None, None, None, Int32(0), Int32(0), Int32(0))]
                 c_pipe = pipeline.PipelineTmaStore.create(
                     num_stages=self.num_c_stage,
                     producer_group=pipeline.CooperativeGroup(pipeline.Agent.Thread, 32 * len(self.epilogue_warp_id))