diff --git a/builder/image-ros.sh b/builder/image-ros.sh
index d849fe364..464fa3de3 100755
--- a/builder/image-ros.sh
+++ b/builder/image-ros.sh
@@ -119,7 +119,8 @@ my_travis_retry apt-get install -y --no-install-recommends \
     ros-${ROS_DISTRO}-cmake-modules \
     ros-${ROS_DISTRO}-image-view \
     ros-${ROS_DISTRO}-nodelet-topic-tools \
-    ros-${ROS_DISTRO}-stereo-msgs
+    ros-${ROS_DISTRO}-stereo-msgs \
+    ros-${ROS_DISTRO}-vision-msgs
 
 # TODO move GeographicLib datasets to Mavros debian package
 echo "--- Install GeographicLib datasets (needed for mavros)" \
diff --git a/builder/test/tests.py b/builder/test/tests.py
index 2af206016..73bf8e669 100755
--- a/builder/test/tests.py
+++ b/builder/test/tests.py
@@ -23,6 +23,10 @@
 from led_msgs.srv import SetLEDs
 from led_msgs.msg import LEDStateArray, LEDState
 from aruco_pose.msg import Marker, MarkerArray, Point2D
+from vision_msgs.msg import BoundingBox2D, BoundingBox2DArray, BoundingBox3D, BoundingBox3DArray, \
+    Classification2D, Classification3D, Detection2D, Detection2DArray, Detection3D, Detection3DArray, \
+    ObjectHypothesis, ObjectHypothesisWithPose, VisionInfo,
+
 from clover import long_callback
 
 import dynamic_reconfigure.client