diff --git a/src/auto_gpt_plugin_template/__init__.py b/src/auto_gpt_plugin_template/__init__.py
index c8340dd..3ea2cb8 100644
--- a/src/auto_gpt_plugin_template/__init__.py
+++ b/src/auto_gpt_plugin_template/__init__.py
@@ -90,14 +90,14 @@ def can_handle_post_planning(self) -> bool:
         return False
 
     @abc.abstractmethod
-    def post_planning(self, response: str) -> str:
+    def post_planning(self, response: Dict[Any, Any]) -> Dict[Any, Any]:
         """This method is called after the planning chat completion is done.
 
         Args:
-            response (str): The response.
+            response (Dict[Any, Any]): The response.
 
         Returns:
-            str: The resulting response.
+            Dict[Any, Any]: The resulting response.
         """
         pass