diff --git a/example/ck_tile/99_toy_example/03_flash_attention_fwd/block_gemm_areg_bsmem_creg_v1.hpp b/example/ck_tile/99_toy_example/03_flash_attention_fwd/block_gemm_areg_bsmem_creg_v1.hpp
index 23ca02946b..fc18640958 100644
--- a/example/ck_tile/99_toy_example/03_flash_attention_fwd/block_gemm_areg_bsmem_creg_v1.hpp
+++ b/example/ck_tile/99_toy_example/03_flash_attention_fwd/block_gemm_areg_bsmem_creg_v1.hpp
@@ -36,7 +36,7 @@ struct BlockGemmARegBSmemCRegV1
         constexpr index_t NPerBlock = BlockGemmShape::kN;
         constexpr index_t KPerBlock = BlockGemmShape::kK;
 
-        constexpr auto config     = Policy::template GetWarpGemmMWarpNWarp<Problem>();
+        constexpr auto config     = Policy::template GetWarpGemmMWarpNWarp<Problem, MPerBlock>();
         using WG                  = remove_cvref_t<decltype(config.template get<0>())>;
         constexpr index_t MPerXDL = WG::kM;
         constexpr index_t NPerXDL = WG::kN;