diff --git a/src/airunner/handlers/llm/agent/mistral_agent.py b/src/airunner/handlers/llm/agent/mistral_agent.py
index f8cc25de3..23639693d 100644
--- a/src/airunner/handlers/llm/agent/mistral_agent.py
+++ b/src/airunner/handlers/llm/agent/mistral_agent.py
@@ -714,7 +714,7 @@ def on_delete_messages_after_id(self, data: Dict):
             if self._chat_engine:
                 self._chat_engine.memory = self._chat_memory
 
-    def _update_system_prompt(self) -> Dict:
+    def _update_system_prompt(self):
         self.chat_engine_tool.update_system_prompt(self._system_prompt)
         self.rag_engine_tool.update_system_prompt(self._rag_system_prompt)