Grab the environment before querying the identities (#176)

Flagsmith · Jul 9, 2021 · 05e2da9 · 05e2da9
1 parent f3d7fb0
commit 05e2da9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/api/environments/identities/views.py b/api/environments/identities/views.py
@@ -27,8 +27,8 @@ class IdentityViewSet(viewsets.ModelViewSet):
     pagination_class = CustomPagination
 
     def get_queryset(self):
-        environment_api_key = self.kwargs["environment_api_key"]
-        queryset = Identity.objects.filter(environment__api_key=environment_api_key)
+        environment = self.get_environment_from_request()
+        queryset = Identity.objects.filter(environment=environment)
 
         search_query = self.request.query_params.get("q")
         if search_query: