Fix base class reference

jazzband · Jun 19, 2019 · 99e426d · 99e426d
1 parent d4dc9b1
commit 99e426d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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'