diff --git a/vision_agent/agent/vision_agent.py b/vision_agent/agent/vision_agent.py
index 9d929204..7fc3cecd 100644
--- a/vision_agent/agent/vision_agent.py
+++ b/vision_agent/agent/vision_agent.py
@@ -323,7 +323,7 @@ def self_reflect(
         question=question, tool_results=str(tool_result), final_answer=final_answer
     )
     if issubclass(type(reflect_model), LMM):
-        return reflect_model(prompt, image=image)
+        return reflect_model(prompt, image=image)  # type: ignore
     return reflect_model(prompt)
 
 
@@ -384,7 +384,7 @@ def chat_with_workflow(
 
         reflections = ""
         final_answer = ""
-        all_tool_results = []
+        all_tool_results: List[Dict] = []
 
         for _ in range(self.max_retries):
             task_list = create_tasks(self.task_model, question, self.tools, reflections)
diff --git a/vision_agent/tools/tools.py b/vision_agent/tools/tools.py
index 1fbbd182..a2b75851 100644
--- a/vision_agent/tools/tools.py
+++ b/vision_agent/tools/tools.py
@@ -385,8 +385,8 @@ class SegArea(Tool):
 
     def __call__(self, masks: Union[str, Path]) -> float:
         pil_mask = Image.open(str(masks))
-        np_mask = np.array(pil_mask)  # type: ignore
-        return round(np.sum(np_mask) / 255, 2)
+        np_mask = np.array(pil_mask)
+        return cast(float, round(np.sum(np_mask) / 255, 2))
 
 
 class Add(Tool):