diff --git a/example/ck_tile/18_hstu_attention/hstu_attention_fwd_pipeline.hpp b/example/ck_tile/18_hstu_attention/hstu_attention_fwd_pipeline.hpp
index 6c3cb1e638..c605e1dfca 100644
--- a/example/ck_tile/18_hstu_attention/hstu_attention_fwd_pipeline.hpp
+++ b/example/ck_tile/18_hstu_attention/hstu_attention_fwd_pipeline.hpp
@@ -84,9 +84,9 @@ struct HstuAttentionFwdPipelineQRKSVS
             else if constexpr(kQKHeaddim == 96 || kQKHeaddim == 128)
             {
                 if constexpr(kHasBias)
-                    return 1;
-                else
                     return 2;
+                else
+                    return 3;
             }
             else if constexpr(kQKHeaddim == 256)
             {