diff --git a/dynamic_stack_decider/package.xml b/dynamic_stack_decider/package.xml
index 198fed1..12336bc 100644
--- a/dynamic_stack_decider/package.xml
+++ b/dynamic_stack_decider/package.xml
@@ -24,6 +24,7 @@
   <depend>rclpy</depend>
   <depend>std_msgs</depend>
   <depend>python3-yaml</depend>
+  <test_depend>python3-pytest</test_depend>
   <test_depend>python3-coverage</test_depend>
 
   <export>
diff --git a/dynamic_stack_decider_visualization/package.xml b/dynamic_stack_decider_visualization/package.xml
index b3dfc4f..62dff5c 100644
--- a/dynamic_stack_decider_visualization/package.xml
+++ b/dynamic_stack_decider_visualization/package.xml
@@ -25,6 +25,7 @@
 
   <depend>python3-pydot</depend>
 
+  <test_depend>python3-pytest</test_depend>
 
   <export>
     <architecture_independent/>
diff --git a/dynamic_stack_decider_visualization/setup.py b/dynamic_stack_decider_visualization/setup.py
index 12ce939..6ec67bf 100644
--- a/dynamic_stack_decider_visualization/setup.py
+++ b/dynamic_stack_decider_visualization/setup.py
@@ -12,6 +12,7 @@
         ("share/" + package_name, ["plugin.xml"]),
     ],
     install_requires=["setuptools"],
+    tests_require=["pytest"],
     zip_safe=True,
     entry_points={
         "console_scripts": [