diff --git a/debian/rules b/debian/rules
index b5ee02f8b8ed23cabb0ffd7875db06b2c130c4b2..968fab70050fd8541734b2582940bb18169d33e0 100755
--- a/debian/rules
+++ b/debian/rules
@@ -1,10 +1,12 @@
 #! /usr/bin/make -f
 
 export PYBUILD_NAME=antlr4
-export PYBUILD_TEST_ARGS=--pattern "*.py" --start-directory test
 
 %:
 	dh $@ --buildsystem=pybuild
 
+override_dh_auto_test:
+	pybuild --test --test-custom --test-args "python{version} test/run.py"
+
 override_dh_installchangelogs:
 	dh_installchangelogs -k RELEASE-4.5.txt
diff --git a/debian/tests/run-unit-test b/debian/tests/run-unit-test
index a0b260445c151f4169ac499d69183b03d3c73459..95a8b8798e41eea4a2643c7ba33460bddb600aab 100644
--- a/debian/tests/run-unit-test
+++ b/debian/tests/run-unit-test
@@ -1,4 +1,4 @@
 #!/bin/bash
 set -e
 
-python3 -m unittest discover --start-directory /usr/lib/python3/dist-packages/antlr4/test/ --pattern "*.py"
+python3 test/run.py