diff --git a/src/planai/llm_interface.py b/src/planai/llm_interface.py
index d867bd8..2aa3005 100644
--- a/src/planai/llm_interface.py
+++ b/src/planai/llm_interface.py
@@ -141,7 +141,7 @@ def generate_pydantic(
             )
 
         if debug_saver is not None:
-            debug_saver(prompt=formatted_prompt, response=raw_response)
+            debug_saver(prompt=formatted_prompt, response=response)
 
         return response