diff --git a/tests/test_schemas.py b/tests/test_schemas.py index 6c02c9d23..614dce246 100644 --- a/tests/test_schemas.py +++ b/tests/test_schemas.py @@ -41,7 +41,7 @@ class ExampleViewSet(ModelViewSet): @detail_route(methods=['post'], serializer_class=AnotherSerializer) def custom_action(self, request, pk): - return super(ExampleSerializer, self).retrieve(self, request) + return super(ExampleViewSet, self).retrieve(self, request) class ExampleView(APIView):