diff --git a/langdspy/prompt_runners.py b/langdspy/prompt_runners.py
index 9eab540..de96481 100644
--- a/langdspy/prompt_runners.py
+++ b/langdspy/prompt_runners.py
@@ -193,6 +193,8 @@ def _invoke_with_retries(self, chain, input, max_tries=1, config: Optional[Runna
                     parsed_output[attr_name] = transformed_val
 
             end_time = time.time()
+
+            # The model named may be called model_name or model depending on version of langchain_anthropic
             try:
                 model_name = config["llm"].model_name
             except AttributeError: