diff --git a/include/group/gemm/impl/unaligned_xmx_xe.hpp b/include/group/gemm/impl/unaligned_xmx_xe.hpp
index 4d0b956e..f48af5d4 100755
--- a/include/group/gemm/impl/unaligned_xmx_xe.hpp
+++ b/include/group/gemm/impl/unaligned_xmx_xe.hpp
@@ -327,11 +327,11 @@ class gemm_t<compute_policy_unaligned_xmx<compute_attr_, perf_tuning_knob_,
         partial_matB_t partial_matB;
         //  >>>>>>>>>>>>>>>>>> pre_processing init
         pre_processing.init(g, args.pre_processing_args);
-        uint64_t base_A = slm_base_a + sg_idy * tile_size_a;
-        uint64_t base_B = slm_base_b + sg_idx * tile_size_b;
+        uint32_t base_A = slm_base_a + sg_idy * tile_size_a;
+        uint32_t base_B = slm_base_b + sg_idx * tile_size_b;
 
-        uint64_t store_idx = 0;
-        uint64_t load_idx = 0;
+        uint32_t store_idx = 0;
+        uint32_t load_idx = 0;
 
         matA_payload_t matA_payload(args.matA_base_desc);
         matA_payload_local_st_t matA_local_st_payload(base_A, tile_size_x_a,
diff --git a/include/subgroup/cooperative_load_helper.hpp b/include/subgroup/cooperative_load_helper.hpp
index d508a7fe..2bbb7263 100755
--- a/include/subgroup/cooperative_load_helper.hpp
+++ b/include/subgroup/cooperative_load_helper.hpp
@@ -42,7 +42,7 @@ class cooperative_load_helper_t<matAcc_t_, mem_layout::row_major,
     static constexpr gpu_arch arch_tag = arch_tag_;
     using matAcc_t = matAcc_t_;
     using dtype = typename matAcc_t::dtype;
-    using tile_desc_t = matAcc_t::tile_desc;
+    using tile_desc_t = typename matAcc_t::tile_desc;
     static constexpr mem_layout layout = mem_layout::row_major;
 
 private:
@@ -101,7 +101,7 @@ class cooperative_load_helper_t<matAcc_t_, mem_layout::col_major,
     static constexpr gpu_arch arch_tag = arch_tag_;
     using matAcc_t = matAcc_t_;
     using dtype = typename matAcc_t::dtype;
-    using tile_desc_t = matAcc_t::tile_desc;
+    using tile_desc_t = typename matAcc_t::tile_desc;
     static constexpr mem_layout layout = mem_layout::col_major;
 
 private: