diff --git a/wagtailmenus/api/v1/tests/test_base_views.py b/wagtailmenus/api/v1/tests/test_base_views.py
index bbc0253af..8a2b2d76a 100644
--- a/wagtailmenus/api/v1/tests/test_base_views.py
+++ b/wagtailmenus/api/v1/tests/test_base_views.py
@@ -1,12 +1,12 @@
 from django.test import override_settings, TestCase
 
-from wagtailmenus.api.v1.views import MenuGeneratorView
+from wagtailmenus.api.v1.views import BaseMenuGeneratorView
 from wagtailmenus.api.v1.serializers import FlatMenuSerializer, MainMenuSerializer
 
 
 class TestGetSerializerClass(TestCase):
 
-    class CustomView(MenuGeneratorView):
+    class CustomView(BaseMenuGeneratorView):
         default_serializer_class = MainMenuSerializer
         serializer_class_setting_name = 'MAIN_MENU_SERIALIZER'