diff --git a/entity/tests/model_tests.py b/entity/tests/model_tests.py
index af51aa1..c9299b2 100644
--- a/entity/tests/model_tests.py
+++ b/entity/tests/model_tests.py
@@ -1125,14 +1125,14 @@ def test_get_all_entities(self):
             [None, account_kind],
         ])
 
-        with self.assertNumQueries(3):
+        with self.assertNumQueries(4):
             membership_cache = EntityGroup.objects.get_membership_cache()
             entities_by_kind = get_entities_by_kind(membership_cache=membership_cache)
 
             for entity_group in entity_groups:
                 entity_group.get_all_entities(membership_cache, entities_by_kind)
 
-        with self.assertNumQueries(3):
+        with self.assertNumQueries(4):
             get_entities_by_kind()
 
         # Make sure to hit the no group cache case