diff --git a/cmstestsuite/unit_tests/schema_diff_test.py b/cmstestsuite/unit_tests/schema_diff_test.py
index 0dd2c2d965..eceb42489d 100644
--- a/cmstestsuite/unit_tests/schema_diff_test.py
+++ b/cmstestsuite/unit_tests/schema_diff_test.py
@@ -35,7 +35,12 @@ def split_schema(schema: str) -> list[list[str]]:
     statements: list[list[str]] = []
     cur_statement: list[str] = []
     for line in schema.splitlines():
-        if line == "" or line.startswith("--"):
+        if (
+            line == ""
+            or line.startswith("--")
+            or line.startswith("\\restrict")
+            or line.startswith("\\unrestrict")
+        ):
             continue
         cur_statement.append(line)
         if line.endswith(";"):