diff --git a/builder/image-ros.sh b/builder/image-ros.sh
index 93a1b2be9..a31334fb7 100755
--- a/builder/image-ros.sh
+++ b/builder/image-ros.sh
@@ -137,7 +137,8 @@ my_travis_retry apt-get install -y --no-install-recommends \
     ros-${ROS_DISTRO}-ws281x \
     ros-${ROS_DISTRO}-rosshow \
     ros-${ROS_DISTRO}-cmake-modules \
-    ros-${ROS_DISTRO}-image-view
+    ros-${ROS_DISTRO}-image-view \
+    ros-${ROS_DISTRO}-image-geometry
 
 # TODO move GeographicLib datasets to Mavros debian package
 echo_stamp "Install GeographicLib datasets (needed for mavros)" \
diff --git a/builder/test/tests.py b/builder/test/tests.py
index bba3dc121..2af206016 100755
--- a/builder/test/tests.py
+++ b/builder/test/tests.py
@@ -33,6 +33,7 @@
 import VL53L1X
 import pymavlink
 from pymavlink import mavutil
+from image_geometry import PinholeCameraModel, StereoCameraModel
 # from espeak import espeak
 from pyzbar import pyzbar
 import docopt