diff --git a/langdspy/prompt_runners.py b/langdspy/prompt_runners.py
index de96481..1bfb6f9 100644
--- a/langdspy/prompt_runners.py
+++ b/langdspy/prompt_runners.py
@@ -200,7 +200,7 @@ def _invoke_with_retries(self, chain, input, max_tries=1, config: Optional[Runna
             except AttributeError:
                 model_name = config["llm"].model
 
-            self.prompt_history.add_entry(config["llm"].model_name, formatted_prompt, res, parsed_output, validation_err, start_time, end_time)
+            self.prompt_history.add_entry(model_name, formatted_prompt, res, parsed_output, validation_err, start_time, end_time)
 
             res = {attr_name: parsed_output.get(attr_name, None) for attr_name in self.template.output_variables.keys()}