diff --git a/notifications/tests/settings.py b/notifications/tests/settings.py
index bd13d81..c5e084b 100644
--- a/notifications/tests/settings.py
+++ b/notifications/tests/settings.py
@@ -20,6 +20,7 @@ DATABASES = {
 MIDDLEWARE_CLASSES = (
     'django.contrib.sessions.middleware.SessionMiddleware',
     'django.contrib.auth.middleware.AuthenticationMiddleware',
+    'django.contrib.messages.middleware.MessageMiddleware'
 )
 
 # Django >= 2.0
@@ -29,6 +30,7 @@ INSTALLED_APPS = (
     'django.contrib.admin',
     'django.contrib.auth',
     'django.contrib.contenttypes',
+    'django.contrib.messages',
     'django.contrib.staticfiles',
     'django.contrib.sessions',
     'notifications.tests',
diff --git a/notifications/tests/urls.py b/notifications/tests/urls.py
index 60c745d..f1ad894 100644
--- a/notifications/tests/urls.py
+++ b/notifications/tests/urls.py
@@ -4,12 +4,22 @@ from distutils.version import StrictVersion  # pylint: disable=no-name-in-module
 
 from django import get_version
 from django.contrib import admin
-from django.contrib.auth.views import login
 from notifications.tests.views import (live_tester,  # pylint: disable=no-name-in-module,import-error
                                        make_notification)
 
-if StrictVersion(get_version()) >= StrictVersion('2.0'):
+if StrictVersion(get_version()) >= StrictVersion('2.1'):
     from django.urls import include, path  # noqa
+    from django.contrib.auth.views import LoginView
+    urlpatterns = [
+        path('test_make/', make_notification),
+        path('test/', live_tester),
+        path('login/', LoginView.as_view(), name='login'),  # reverse for django login is not working
+        path('admin/', admin.site.urls),
+        path('', include('notifications.urls', namespace='notifications')),
+    ]
+elif StrictVersion(get_version()) >= StrictVersion('2.0') and StrictVersion(get_version()) < StrictVersion('2.1'):
+    from django.urls import include, path  # noqa
+    from django.contrib.auth.views import login
     urlpatterns = [
         path('test_make/', make_notification),
         path('test/', live_tester),
@@ -19,6 +29,7 @@ if StrictVersion(get_version()) >= StrictVersion('2.0'):
     ]
 else:
     from django.conf.urls import include, url
+    from django.contrib.auth.views import login
     urlpatterns = [
         url(r'^login/$', login, name='login'),  # reverse for django login is not working
         url(r'^test_make/', make_notification),