diff --git a/include/ck_tile/ops/flatmm/kernel/flatmm_kernel.hpp b/include/ck_tile/ops/flatmm/kernel/flatmm_kernel.hpp
index ac682ba04c..b731a9a5bc 100755
--- a/include/ck_tile/ops/flatmm/kernel/flatmm_kernel.hpp
+++ b/include/ck_tile/ops/flatmm/kernel/flatmm_kernel.hpp
@@ -53,10 +53,12 @@ struct FlatmmScalePointer
         }
         else if constexpr(granularity == 1)
         {
+            ret.ptr = ptr + offset;
             ret.length = length - offset;
         }
         else
         {
+            ret.ptr = ptr + offset / granularity;
             ret.length = length - offset / granularity;
         }
         return ret;