diff --git a/debian/control b/debian/control
index 43b2ce4..64a78b8 100644
--- a/debian/control
+++ b/debian/control
@@ -12,6 +12,7 @@ Build-Depends: debhelper-compat (= 13),
                python3-flake8,
                python3-jsonschema,
                python3-nose,
+               python3-parameterized,
                python3-pyudev,
                python3-setuptools,
 Standards-Version: 4.5.0
diff --git a/debian/probert-common.install b/debian/probert-common.install
index b4706c6..29ad1ac 100644
--- a/debian/probert-common.install
+++ b/debian/probert-common.install
@@ -7,15 +7,6 @@ usr/lib/python3.*/dist-packages/probert/__init__.py
 usr/lib/python3.*/dist-packages/probert/log.py
 usr/lib/python3.*/dist-packages/probert/prober.py
 usr/lib/python3.*/dist-packages/probert/tests/__init__.py
-usr/lib/python3.*/dist-packages/probert/tests/data/dasdd.view
-usr/lib/python3.*/dist-packages/probert/tests/data/dasde.view
-usr/lib/python3.*/dist-packages/probert/tests/data/fake_probe_all.json
-usr/lib/python3.*/dist-packages/probert/tests/fakes.py
-usr/lib/python3.*/dist-packages/probert/tests/helpers.py
-usr/lib/python3.*/dist-packages/probert/tests/test_dasd.py
-usr/lib/python3.*/dist-packages/probert/tests/test_lvm.py
-usr/lib/python3.*/dist-packages/probert/tests/test_multipath.py
-usr/lib/python3.*/dist-packages/probert/tests/test_prober.py
-usr/lib/python3.*/dist-packages/probert/tests/test_storage.py
-usr/lib/python3.*/dist-packages/probert/tests/test_utils.py
+usr/lib/python3.*/dist-packages/probert/tests/data/*
+usr/lib/python3.*/dist-packages/probert/tests/*.py
 usr/lib/python3.*/dist-packages/probert/utils.py
diff --git a/debian/probert-storage.install b/debian/probert-storage.install
index edf357c..2e6ad76 100644
--- a/debian/probert-storage.install
+++ b/debian/probert-storage.install
@@ -5,6 +5,8 @@ usr/lib/python3.*/dist-packages/probert/filesystem.py
 usr/lib/python3.*/dist-packages/probert/lvm.py
 usr/lib/python3.*/dist-packages/probert/mount.py
 usr/lib/python3.*/dist-packages/probert/multipath.py
+usr/lib/python3.*/dist-packages/probert/nvme.py
+usr/lib/python3.*/dist-packages/probert/os.py
 usr/lib/python3.*/dist-packages/probert/raid.py
 usr/lib/python3.*/dist-packages/probert/storage.py
 usr/lib/python3.*/dist-packages/probert/zfs.py