diff --git a/include/ck_tile/core/tensor/tile_distribution.hpp b/include/ck_tile/core/tensor/tile_distribution.hpp
index 8fe16a1693..606b346e19 100644
--- a/include/ck_tile/core/tensor/tile_distribution.hpp
+++ b/include/ck_tile/core/tensor/tile_distribution.hpp
@@ -114,8 +114,7 @@ struct tile_distribution
         }
         else if constexpr(NDimP == 2)
         {
-            return array<index_t, 2>{get_warp_id(bool_constant<save_warp_id_in_sgpr>{}),
-                                     get_lane_id()};
+            return array<index_t, 2>{get_warp_id(), get_lane_id()};
         }
     }