diff --git a/awx/main/routing.py b/awx/main/routing.py
index 290e202afa..239db32fad 100644
--- a/awx/main/routing.py
+++ b/awx/main/routing.py
@@ -4,9 +4,10 @@ import logging
 from django.conf import settings
 from django.urls import re_path
 
-from channels.auth import AuthMiddlewareStack
 from channels.routing import ProtocolTypeRouter, URLRouter
 
+from ansible_base.lib.channels.middleware import DrfAuthMiddlewareStack
+
 from . import consumers
 
 
@@ -34,6 +35,6 @@ websocket_urlpatterns = [
 
 application = AWXProtocolTypeRouter(
     {
-        'websocket': AuthMiddlewareStack(URLRouter(websocket_urlpatterns)),
+        'websocket': DrfAuthMiddlewareStack(URLRouter(websocket_urlpatterns)),
     }
 )