diff --git a/example/ck_tile/18_hstu_attention/hstu_attention_fwd_kernel.hpp b/example/ck_tile/18_hstu_attention/hstu_attention_fwd_kernel.hpp
index 324916bf77..c07e792b26 100644
--- a/example/ck_tile/18_hstu_attention/hstu_attention_fwd_kernel.hpp
+++ b/example/ck_tile/18_hstu_attention/hstu_attention_fwd_kernel.hpp
@@ -624,7 +624,7 @@ struct HstuAttentionFwdKernel
             make_tile_window(q_dram,
                              [&]() {
                                  return make_tuple(number<HstuAttentionPipeline::kM0>{},
-                                                   number<HstuAttentionPipeline::kQKHeaddim>{});
+                                                   number<HstuAttentionPipeline::kSubQKHeaddim>{});
                              }(),
                              {i_m0, 0});