diff --git a/package.xml b/package.xml
index 34c692c..66bf344 100644
--- a/package.xml
+++ b/package.xml
@@ -19,5 +19,7 @@
   <run_depend>opencv_candidate</run_depend>
   <run_depend>cv_backports</run_depend>
 
+  <test_depend>unittest</test_depend>
+
   <buildtool_depend>catkin</buildtool_depend>
 </package>
diff --git a/test/doc_opencv.py b/test/doc_opencv.py
index 23a4523..71eed6f 100755
--- a/test/doc_opencv.py
+++ b/test/doc_opencv.py
@@ -14,5 +14,5 @@ def test_list_modules(self):
         ecto.list_ecto_module(imgproc)
 
 if __name__ == '__main__':
-    import rostest
-    rostest.rosrun('ecto_opencv', 'test_ecto_opencv', TestListModules)
+    import rosunit
+    rosunit.rosrun('ecto_opencv', 'test_ecto_opencv', TestListModules)
diff --git a/test/test_image_saver.py b/test/test_image_saver.py
index efd0653..a4e0355 100755
--- a/test/test_image_saver.py
+++ b/test/test_image_saver.py
@@ -27,5 +27,5 @@ def test_image_saver(self):
         os.remove('img_00000.png')
 
 if __name__ == '__main__':
-    import rostest
-    rostest.rosrun('ecto_opencv', 'test_image_saver', TestImageSaver)
+    import rosunit
+    rosunit.unitrun('ecto_opencv', 'test_image_saver', TestImageSaver)