diff --git a/utils/test/test-ptrace_parse.py b/utils/test/test-ptrace_parse.py
index d7dea4790..b83da5031 100644
--- a/utils/test/test-ptrace_parse.py
+++ b/utils/test/test-ptrace_parse.py
@@ -9,27 +9,29 @@
 #
 # ------------------------------------------------------------------
 
-import apparmor.aa as aa
 import unittest
-from common_test import AAParseTest, setup_regex_tests
+from common_test import AATest, setup_all_loops
 
-class AAParsePtraceTest(AAParseTest):
-    def setUp(self):
-        self.parse_function = aa.parse_ptrace_rule
+from apparmor.rule.ptrace import PtraceRule
+
+class AAParsePtraceTest(AATest):
+    def _run_test(self, params, expected):
+        rule_obj = PtraceRule.parse(params)
+        self.assertEqual(rule_obj.get_clean(), expected)
 
     tests = [
-        ('ptrace,', 'ptrace base keyword rule'),
-        ('ptrace (readby),', 'ptrace readby rule'),
-        ('ptrace (trace),', 'ptrace trace rule'),
-        ('ptrace (trace read),', 'ptrace multi-perm rule'),
-        ('ptrace r,', 'ptrace r rule'),
-        ('ptrace w,', 'ptrace w rule'),
-        ('ptrace rw,', 'ptrace rw rule'),
-        ('ptrace read peer=foo,', 'ptrace peer rule 1'),
-        ('ptrace (trace read) peer=/usr/bin/bar,', 'ptrace peer rule 2'),
-        ('ptrace wr peer=/sbin/baz,', 'ptrace peer rule 3'),
+        ('ptrace,',                 'ptrace,'),
+        ('ptrace (readby),',        'ptrace readby,'),
+        ('ptrace (trace),',         'ptrace trace,'),
+        ('ptrace (trace read),',    'ptrace (read trace),'),
+        ('ptrace r,',               'ptrace r,'),
+        ('ptrace w,',               'ptrace w,'),
+        ('ptrace rw,',              'ptrace rw,'),
+        ('ptrace read peer=foo,',   'ptrace read peer=foo,'),
+        ('ptrace (trace read) peer=/usr/bin/bar,', 'ptrace (read trace) peer=/usr/bin/bar,'),
+        ('ptrace wr peer=/sbin/baz,',   'ptrace wr peer=/sbin/baz,'),
     ]
 
+setup_all_loops(__name__)
 if __name__ == '__main__':
-    setup_regex_tests(AAParsePtraceTest)
     unittest.main(verbosity=2)
diff --git a/utils/test/test-regex_matches.py b/utils/test/test-regex_matches.py
index 8d1efb00d..3059ebf19 100644
--- a/utils/test/test-regex_matches.py
+++ b/utils/test/test-regex_matches.py
@@ -14,7 +14,7 @@ import unittest
 from common_test import AATest, setup_all_loops
 from apparmor.common import AppArmorBug, AppArmorException
 
-from apparmor.regex import strip_quotes, parse_profile_start_line, re_match_include, RE_PROFILE_START, RE_PROFILE_CAP, RE_PROFILE_SIGNAL
+from apparmor.regex import strip_quotes, parse_profile_start_line, re_match_include, RE_PROFILE_START, RE_PROFILE_CAP, RE_PROFILE_PTRACE, RE_PROFILE_SIGNAL
 
 
 class AARegexTest(AATest):
@@ -318,7 +318,7 @@ class AARegexPtrace(AARegexTest):
     '''Tests for RE_PROFILE_PTRACE'''
 
     def AASetup(self):
-        self.regex = aa.RE_PROFILE_PTRACE
+        self.regex = RE_PROFILE_PTRACE
 
     tests = [
         #                                            audit      allow  rule                                     rule details                    comment