diff --git a/pylint_django/tests/input/func_model_no_explicit_unicode_str_compat.py b/pylint_django/tests/input/func_model_no_explicit_unicode_str_compat.py
index 9eba773a..764a1b37 100644
--- a/pylint_django/tests/input/func_model_no_explicit_unicode_str_compat.py
+++ b/pylint_django/tests/input/func_model_no_explicit_unicode_str_compat.py
@@ -6,7 +6,7 @@
 #  pylint: disable=missing-docstring
 
 from django.db import models
-from django.utils.encoding import python_2_unicode_compatible
+from six import python_2_unicode_compatible
 
 
 @python_2_unicode_compatible
diff --git a/pylint_django/tests/input/func_noerror_unicode_py2_compatible.py b/pylint_django/tests/input/func_noerror_unicode_py2_compatible.py
index eb74837c..596c667c 100644
--- a/pylint_django/tests/input/func_noerror_unicode_py2_compatible.py
+++ b/pylint_django/tests/input/func_noerror_unicode_py2_compatible.py
@@ -5,7 +5,7 @@
 See https://github.com/PyCQA/pylint-django/issues/10
 """
 #  pylint: disable=missing-docstring
-from django.utils.encoding import python_2_unicode_compatible
+from six import python_2_unicode_compatible
 from django.db import models