diff --git a/examples/custom_tools/run_custom_tool.py b/examples/custom_tools/run_custom_tool.py
index 593b4959..beaa9eca 100644
--- a/examples/custom_tools/run_custom_tool.py
+++ b/examples/custom_tools/run_custom_tool.py
@@ -1,9 +1,9 @@
+from template_match import template_matching_with_rotation
+
 import vision_agent as va
 from vision_agent.image_utils import get_image_size, normalize_bbox
 from vision_agent.tools import Tool, register_tool
 
-from template_match import template_matching_with_rotation
-
 
 @register_tool
 class TemplateMatch(Tool):
diff --git a/vision_agent/tools/tools.py b/vision_agent/tools/tools.py
index 64f97467..a3dab630 100644
--- a/vision_agent/tools/tools.py
+++ b/vision_agent/tools/tools.py
@@ -905,7 +905,7 @@ def __call__(self, image: str) -> dict:
             raise ValueError(f"Request failed: {res.text}")
 
         data = res.json()
-        output = {"labels": [], "bboxes": [], "scores": []}
+        output: Dict[str, List] = {"labels": [], "bboxes": [], "scores": []}
         for det in data[0]:
             output["labels"].append(det["text"])
             box = [