diff --git a/kdb-bot/src/bot_graphql/abc/query_abc.py b/kdb-bot/src/bot_graphql/abc/query_abc.py index a90ccfd3..f389e4c7 100644 --- a/kdb-bot/src/bot_graphql/abc/query_abc.py +++ b/kdb-bot/src/bot_graphql/abc/query_abc.py @@ -63,7 +63,7 @@ class QueryABC(ObjectType): return self._resolve_collection(collection, *args, **kwargs) self.set_field(f"{name}s", wrapper) - self.set_field(f"{name}Count", lambda *args: get_collection(*args).count()) + self.set_field(f"{name}Count", lambda *args: wrapper(*args).count()) @ServiceProviderABC.inject def _can_user_see_element(self, user: AuthUser, element, services: ServiceProviderABC) -> bool: