diff --git a/layers/loader.py b/layers/loader.py
index 07eb8a4..84c2322 100644
--- a/layers/loader.py
+++ b/layers/loader.py
@@ -1,10 +1,10 @@
 import os
 import sys
+from importlib import import_module
 
-from django.template.base import TemplateDoesNotExist
+from django.template import TemplateDoesNotExist
 from django.template.loaders.app_directories import Loader as BaseLoader
 from django.conf import settings
-from django.utils.importlib import import_module
 from django.core.exceptions import ImproperlyConfigured
 
 
diff --git a/layers/middleware.py b/layers/middleware.py
index 2054ce4..6f7f2b6 100644
--- a/layers/middleware.py
+++ b/layers/middleware.py
@@ -3,7 +3,7 @@
 except ImportError:
     from django.utils._threading_local import local
 
-from django.utils.importlib import import_module
+from importlib import import_module
 from django.conf import settings
 
 _thread_locals = local()