diff --git a/vision_agent/agent/vision_agent.py b/vision_agent/agent/vision_agent.py
index 3928ea71..8627b06c 100644
--- a/vision_agent/agent/vision_agent.py
+++ b/vision_agent/agent/vision_agent.py
@@ -593,10 +593,12 @@ def chat_with_workflow(
         )
 
         if visualize_output:
-            viz_images = all_tool_results[-1]["visualized_output"]
+            viz_images: Sequence[Union[str, Path]] = all_tool_results[-1][
+                "visualized_output"
+            ]
             self._report_visualization_via_callback(viz_images)
-            for image in viz_images:
-                Image.open(image).show()
+            for img in viz_images:
+                Image.open(img).show()
 
         return final_answer, all_tool_results