diff --git a/vision_agent/tools/__init__.py b/vision_agent/tools/__init__.py
index 450c0b26..76045ec1 100644
--- a/vision_agent/tools/__init__.py
+++ b/vision_agent/tools/__init__.py
@@ -11,6 +11,8 @@
     GroundingDINO,
     GroundingSAM,
     ImageCaption,
+    ZeroShotCounting,
+    VisualPromptCounting,
     SegArea,
     SegIoU,
     Tool,