diff --git a/comfyui.py b/comfyui.py
index 8fe9a96..425fb55 100644
--- a/comfyui.py
+++ b/comfyui.py
@@ -629,7 +629,7 @@ def _validate_comfyui_args(self, workflow, tool):
                     raise Exception(f"Remap parameter {key} is missing original choices: {choices}")
                                 
     def _inject_args_into_workflow(self, workflow, tool, args):
-
+        base_model = "unknown"
         # Helper function to validate and normalize URLs
         def validate_url(url):
             if not isinstance(url, str):