diff --git a/feedler/endpoints.py b/feedler/endpoints.py
index 7c48447..fb13951 100644
--- a/feedler/endpoints.py
+++ b/feedler/endpoints.py
@@ -1,6 +1,6 @@
 # -*- coding: utf-8 -*-
 
-from wagtail.api.v2.endpoints import BaseAPIEndpoint
+from wagtail.api.v2.views import BaseAPIViewSet
 from wagtail.api.v2.serializers import BaseSerializer
 from wagtail.api.v2.filters import FieldsFilter, OrderingFilter, SearchFilter
 from wagtail.api.v2.pagination import WagtailPagination
@@ -10,7 +10,7 @@ from .models import Entry
 class EntrySerializer(BaseSerializer):
     pass
 
-class EntriesAPIEndpoint(BaseAPIEndpoint):
+class EntriesAPIEndpoint(BaseAPIViewSet):
     base_serializer_class = EntrySerializer
     filter_backends = [FieldsFilter, OrderingFilter, SearchFilter]
     extra_api_fields = [