diff --git a/utils/test/common_test.py b/utils/test/common_test.py
index 759663563..89846bebe 100755
--- a/utils/test/common_test.py
+++ b/utils/test/common_test.py
@@ -51,10 +51,9 @@ class AAParseTest(unittest.TestCase):
             'parse object %s returned "%s", expected "%s"' \
             %(self.parse_function.__doc__, parsed.serialize(), rule))
 
-
-def setup_all_tests():
+def setup_all_loops(module_name):
     '''call setup_tests_loop() for each class in module_name'''
-    for name, obj in inspect.getmembers(sys.modules['__main__']):
+    for name, obj in inspect.getmembers(sys.modules[module_name]):
         if inspect.isclass(obj):
             if issubclass(obj, unittest.TestCase):
                 setup_tests_loop(obj)
diff --git a/utils/test/test-aa.py b/utils/test/test-aa.py
index 3987a3d79..9284806ac 100644
--- a/utils/test/test-aa.py
+++ b/utils/test/test-aa.py
@@ -10,7 +10,7 @@
 # ------------------------------------------------------------------
 
 import unittest
-from common_test import AATest, setup_all_tests
+from common_test import AATest, setup_all_loops
 import os
 import shutil
 import tempfile
@@ -536,6 +536,6 @@ class AaTestInvalid_serialize_parse_profile_start(AATest):
             serialize_parse_profile_start(line, 'somefile', 1, profile, hat, prof_data_profile, prof_data_external, True)
 
 
+setup_all_loops(__name__)
 if __name__ == '__main__':
-    setup_all_tests()
     unittest.main(verbosity=2)
diff --git a/utils/test/test-example.py b/utils/test/test-example.py
index c90ffad03..1624cf887 100644
--- a/utils/test/test-example.py
+++ b/utils/test/test-example.py
@@ -10,7 +10,7 @@
 # ------------------------------------------------------------------
 
 import unittest
-from common_test import AATest, setup_all_tests
+from common_test import AATest, setup_all_loops
 
 class TestFoo(AATest):
     tests = [
@@ -40,6 +40,6 @@ class TestBaz(AATest):
 
 
 
+setup_all_loops(__name__)
 if __name__ == '__main__':
-    setup_all_tests()
     unittest.main(verbosity=2)
diff --git a/utils/test/test-regex_matches.py b/utils/test/test-regex_matches.py
index a35a0dd81..89dea8072 100644
--- a/utils/test/test-regex_matches.py
+++ b/utils/test/test-regex_matches.py
@@ -11,7 +11,7 @@
 
 import apparmor.aa as aa
 import unittest
-from common_test import AATest, setup_all_tests
+from common_test import AATest, setup_all_loops
 from apparmor.common import AppArmorBug
 
 from apparmor.regex import strip_quotes, parse_profile_start_line, RE_PROFILE_START
@@ -19,7 +19,7 @@ from apparmor.regex import strip_quotes, parse_profile_start_line, RE_PROFILE_ST
 
 class AARegexTest(AATest):
     def _run_test(self, params, expected):
-        return regex_test(self, params, expected)
+        return _regex_test(self, params, expected)
 
 class AANamedRegexTest(AATest):
     def _run_test(self, line, expected):
@@ -173,7 +173,7 @@ def setup_split_comment_testcases():
         setattr(AARegexSplitComment, 'test_split_comment_%d' % (i), stub_test)
 
 
-def regex_test(self, line, expected):
+def _regex_test(self, line, expected):
     '''Run a line through self.regex.search() and verify the result
 
     Keyword arguments:
@@ -484,10 +484,10 @@ class TestStripQuotes(AATest):
 
 
 
+setup_all_loops(__name__)
 if __name__ == '__main__':
     # these two are not converted to a tests[] loop yet
     setup_has_comma_testcases()
     setup_split_comment_testcases()
 
-    setup_all_tests()
     unittest.main(verbosity=2)