diff --git a/autogpt/llm/chat.py b/autogpt/llm/chat.py
index 8f7f7d50fb38..a59fe73e9dbf 100644
--- a/autogpt/llm/chat.py
+++ b/autogpt/llm/chat.py
@@ -200,7 +200,7 @@ def chat_with_ai(
                 if not plugin.can_handle_on_planning():
                     continue
                 plugin_response = plugin.on_planning(
-                    agent.prompt_generator, current_context
+                    agent.config.prompt_generator, current_context
                 )
                 if not plugin_response or plugin_response == "":
                     continue