diff --git a/example/ck_tile/01_fmha/fmha_fwd.cpp b/example/ck_tile/01_fmha/fmha_fwd.cpp
index 4b90fa9a50..d7beb7eb07 100644
--- a/example/ck_tile/01_fmha/fmha_fwd.cpp
+++ b/example/ck_tile/01_fmha/fmha_fwd.cpp
@@ -771,7 +771,7 @@ bool run(const ck_tile::ArgParser& arg_parser)
                                                                         : rope_enum::half_rotated)
                                                : rope_enum::none);
         }
-        else
+        else // fmha_fwd_traits or fmha_splitkv_traits
         {
             traits.mask_type           = mask.type;
             traits.bias_type           = bias.type;