diff --git a/memoryscope/memory/worker/summary/get_observation_with_time_worker.py b/memoryscope/memory/worker/summary/get_observation_with_time_worker.py
index 18a77380..3b451a56 100644
--- a/memoryscope/memory/worker/summary/get_observation_with_time_worker.py
+++ b/memoryscope/memory/worker/summary/get_observation_with_time_worker.py
@@ -2,7 +2,7 @@ from typing import List
 
 from memoryscope.constants.common_constants import NEW_OBS_WITH_TIME_NODES
 from memoryscope.constants.language_constants import COLON_WORD
-from memoryscope.memory.worker.write.get_observation_worker import GetObservationWorker
+from memoryscope.memory.worker.summary.get_observation_worker import GetObservationWorker
 from memoryscope.scheme.message import Message
 from memoryscope.utils.datetime_handler import DatetimeHandler
 from memoryscope.utils.tool_functions import prompt_to_msg
diff --git a/memoryscope/memory/worker/summary/load_memory_worker.py b/memoryscope/memory/worker/summary/load_memory_worker.py
index 24972900..1f5e4233 100644
--- a/memoryscope/memory/worker/summary/load_memory_worker.py
+++ b/memoryscope/memory/worker/summary/load_memory_worker.py
@@ -55,7 +55,7 @@ class LoadMemoryWorker(MemoryBaseWorker):
         self.memory_handler.set_memories(NOT_UPDATED_NODES, nodes)
 
     @timer
-    def retrieve_insight_memory(self, query: str):
+    def retrieve_insight_memory(self):
         """
         Retrieves top-K insight memories based on the query and stores them in the memory handler.
         """