diff --git a/vision_agent/agent/vision_agent.py b/vision_agent/agent/vision_agent.py
index 80d56a3e..1aed604f 100644
--- a/vision_agent/agent/vision_agent.py
+++ b/vision_agent/agent/vision_agent.py
@@ -431,7 +431,7 @@ def __init__(
             OpenAILLM(temperature=0.1) if answer_model is None else answer_model
         )
         self.reflect_model = (
-            OpenAILMM(temperature=0.0) if reflect_model is None else reflect_model
+            OpenAILMM(temperature=0.1) if reflect_model is None else reflect_model
         )
         self.max_retries = max_retries
 
diff --git a/vision_agent/llm/llm.py b/vision_agent/llm/llm.py
index 47bd4185..e97bcdeb 100644
--- a/vision_agent/llm/llm.py
+++ b/vision_agent/llm/llm.py
@@ -48,7 +48,7 @@ def generate(self, prompt: str) -> str:
             messages=[
                 {"role": "user", "content": prompt},
             ],
-            **self.kwargs,  # type: ignore
+            **self.kwargs,
         )
 
         return cast(str, response.choices[0].message.content)