diff --git a/src/memos/multi_mem_cube/single_cube.py b/src/memos/multi_mem_cube/single_cube.py
index 1ddd2b1b7..f9e084347 100644
--- a/src/memos/multi_mem_cube/single_cube.py
+++ b/src/memos/multi_mem_cube/single_cube.py
@@ -389,6 +389,7 @@ def _search_pref(
                 top_k=search_req.pref_top_k,
                 info={
                     "user_id": search_req.user_id,
+                    "mem_cube_id": user_context.mem_cube_id,
                     "session_id": search_req.session_id,
                     "chat_history": search_req.chat_history,
                 },
@@ -566,7 +567,7 @@ def _process_pref_mem(
                 message_item_pref = ScheduleMessageItem(
                     user_id=add_req.user_id,
                     session_id=target_session_id,
-                    mem_cube_id=self.cube_id,
+                    mem_cube_id=user_context.mem_cube_id,
                     mem_cube=self.naive_mem_cube,
                     label=PREF_ADD_LABEL,
                     content=json.dumps(messages_list),
@@ -591,7 +592,7 @@ def _process_pref_mem(
                     **(add_req.info or {}),
                     "user_id": add_req.user_id,
                     "session_id": target_session_id,
-                    "mem_cube_id": self.cube_id,
+                    "mem_cube_id": user_context.mem_cube_id,
                 },
             )
             pref_ids_local: list[str] = self.naive_mem_cube.pref_mem.add(pref_memories_local)