diff --git a/vision_agent/tools/tools.py b/vision_agent/tools/tools.py
index 164ca6ea..51ae01ca 100644
--- a/vision_agent/tools/tools.py
+++ b/vision_agent/tools/tools.py
@@ -301,6 +301,7 @@ def __call__(self, prompt: str, image: Union[str, ImageType]) -> Dict:
                 for mask in data["masks"]
             ]
         ret_pred["labels"] = data["labels"]
+        ret_pred["scores"] = data["scores"]
         return ret_pred