diff --git a/src/sql/engine/basic/ob_temp_block_store.h b/src/sql/engine/basic/ob_temp_block_store.h
index 9c16ba8c6..61cfc4183 100644
--- a/src/sql/engine/basic/ob_temp_block_store.h
+++ b/src/sql/engine/basic/ob_temp_block_store.h
@@ -396,7 +396,7 @@ public:
   inline int64_t get_file_size() const { return file_size_; }
   inline int64_t get_max_blk_size() const { return max_block_size_; }
   inline int64_t get_max_hold_mem() const { return max_hold_mem_; }
-  inline ObIAllocator& get_inner_allocator() { return inner_allocator_; }
+  inline common::DefaultPageAllocator& get_inner_allocator() { return inner_allocator_; }
   inline int64_t has_dumped() const { return block_cnt_on_disk_ > 0; }
   inline int64_t get_last_buffer_mem_size() const
   {
diff --git a/src/storage/ddl/ob_direct_load_struct.cpp b/src/storage/ddl/ob_direct_load_struct.cpp
index 62f2b071a..aa9913bd7 100644
--- a/src/storage/ddl/ob_direct_load_struct.cpp
+++ b/src/storage/ddl/ob_direct_load_struct.cpp
@@ -525,7 +525,7 @@ int ObChunkSliceStore::prepare_datum_stores(const uint64_t tenant_id, ObTabletHa
             LOG_WARN("failed to init chunk datum store", K(ret));
           } else {
             datum_store->set_dir_id(dir_id);
-            datum_store->set_allocator(allocator);
+            datum_store->get_inner_allocator().set_tenant_id(tenant_id);
             LOG_INFO("set dir id", K(dir_id));
           }
           if (OB_SUCC(ret)) {