diff --git a/tests/unit/test_refinement_three_level_multipatch.py b/tests/unit/test_refinement_three_level_multipatch.py
index 38aa0cc..2268c5a 100644
--- a/tests/unit/test_refinement_three_level_multipatch.py
+++ b/tests/unit/test_refinement_three_level_multipatch.py
@@ -311,8 +311,6 @@ class ConstructorAndSchedule(unittest.TestCase):
                 mock_backend(**options)
 
 
-if __name__=='__main__':
-    unittest.main()
 
 
 class InitialLevels(unittest.TestCase):
@@ -330,3 +328,7 @@ class InitialLevels(unittest.TestCase):
                      ((), a, None, None), ((), None, b, c)):
             with self.subTest(args=args), self.assertRaises(TypeError):
                 f(*args)
+
+
+if __name__=='__main__':
+    unittest.main()