diff --git a/apps/account/views/back.py b/apps/account/views/back.py index 1a411e9b..1a3f5ccd 100644 --- a/apps/account/views/back.py +++ b/apps/account/views/back.py @@ -24,8 +24,9 @@ class RoleTabRetrieveView(generics.GenericAPIView): """Overridden get_queryset method.""" additional_filters = {} - if self.request.user.userrole_set.country_admin_role().exists(): - additional_filters.update({'userrole__country__code': self.request.country_code}) + if (self.request.user.userrole_set.country_admin_role().exists() and + hasattr(self.request, 'country_code')): + additional_filters.update({'country__code': self.request.country_code}) return models.Role.objects.filter(**additional_filters)\ .annotate_role_name()\