From 3d84c4a78fe6349a3b536c8a78b8dd4bd9bb56e5 Mon Sep 17 00:00:00 2001 From: Diogo Marques Date: Fri, 1 Feb 2019 14:36:48 +0000 Subject: [PATCH] Hotfix: missing import --- wagtail_modeltranslation/tests/tests.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/wagtail_modeltranslation/tests/tests.py b/wagtail_modeltranslation/tests/tests.py index 192d805..e770c80 100755 --- a/wagtail_modeltranslation/tests/tests.py +++ b/wagtail_modeltranslation/tests/tests.py @@ -12,6 +12,10 @@ from django.test.client import RequestFactory from django.test.utils import override_settings from django.utils.translation import get_language, trans_real from modeltranslation import settings as mt_settings, translator +try: + from wagtail.snippets.views.snippets import get_snippet_edit_handler +except ImportError: + from wagtail.wagtailsnippets.views.snippets import get_snippet_edit_handler from wagtail import VERSION from .util import page_factory