diff --git a/graphiql_debug_toolbar/middleware.py b/graphiql_debug_toolbar/middleware.py
index f08117b..7207a26 100644
--- a/graphiql_debug_toolbar/middleware.py
+++ b/graphiql_debug_toolbar/middleware.py
@@ -4,7 +4,11 @@
 from django.template.loader import render_to_string
 from django.utils.encoding import force_str
 
-from debug_toolbar.middleware import _HTML_TYPES
+try:
+    from debug_toolbar.middleware import _HTML_TYPES
+except ImportError:
+    from debug_toolbar.utils import _HTML_TYPES
+
 from debug_toolbar.middleware import DebugToolbarMiddleware as BaseMiddleware
 from debug_toolbar.toolbar import DebugToolbar
 from graphene_django.views import GraphQLView