diff --git a/app/models/user.py b/app/models/user.py index 764f13470..dec7c9692 100644 --- a/app/models/user.py +++ b/app/models/user.py @@ -528,12 +528,13 @@ class AnonymousUser(AnonymousUserMixin): class Users(Sequence): client = user_api_client.get_users_for_service + model = User def __init__(self, service_id): self.users = self.client(service_id) def __getitem__(self, index): - return User(self.users[index]) + return self.model(self.users[index]) def __len__(self): return len(self.users) @@ -557,9 +558,6 @@ class InvitedUsers(Users): if user['status'] != 'accepted' ] - def __getitem__(self, index): - return self.model(self.users[index]) - class OrganisationInvitedUsers(InvitedUsers): client = org_invite_api_client.get_invites_for_organisation