diff --git a/TWB/asgi.py b/TWB/asgi.py
index ad656f2..124638d 100644
--- a/TWB/asgi.py
+++ b/TWB/asgi.py
@@ -22,9 +22,11 @@ os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'TWB.settings')
 
 application = ProtocolTypeRouter({
     'http': get_asgi_application(),
-    "websocket": QueryAuthMiddleware(
-        URLRouter(
-            websocket_urlpatterns
+    "websocket": AllowedHostsOriginValidator(
+        QueryAuthMiddleware(
+            URLRouter(
+                websocket_urlpatterns
+            )
         )
     ),
     # 'websocket': AuthMiddlewareStack(