diff --git a/mm_agents/agent.py b/mm_agents/agent.py
index 9ab64e8..d947fb4 100644
--- a/mm_agents/agent.py
+++ b/mm_agents/agent.py
@@ -1077,7 +1077,10 @@ class PromptAgent:
 
             return actions
 
-    def reset(self):
+    def reset(self, _logger=None):
+        global logger
+        logger = _logger if _logger is not None else logging.getLogger("desktopenv.agent")
+
         self.thoughts = []
         self.actions = []
         self.observations = []