diff --git a/tests/migrations/test_loader.py b/tests/migrations/test_loader.py
index 564e5207b8..f661a03b02 100644
--- a/tests/migrations/test_loader.py
+++ b/tests/migrations/test_loader.py
@@ -86,7 +86,7 @@ class LoaderTests(TestCase):
         )
         # Now render it out!
         project_state = migration_loader.graph.project_state(("migrations", "0001_initial"))
-        self.assertEqual(len(project_state.models), 4)
+        self.assertEqual(len([m for a, m in project_state.models if a == "migrations"]), 1)
 
         book_state = project_state.models["migrations", "book"]
         self.assertEqual(