diff --git a/ee/assist/utils/wsServer.js b/ee/assist/utils/wsServer.js
index f0ea61c8b..3d99d16a0 100644
--- a/ee/assist/utils/wsServer.js
+++ b/ee/assist/utils/wsServer.js
@@ -94,10 +94,6 @@ const createSocketIOServer = function (server, prefix) {
             credentials: true
         },
         path: (prefix ? prefix : '') + '/socket',
-        allowRequest: (req, callback) => {
-            req.res.setHeader('x-host-id', process.env.HOSTNAME || 'unknown');
-            callback(null, true);
-        },
         ...getCompressionConfig()
     };
 
@@ -108,6 +104,10 @@ const createSocketIOServer = function (server, prefix) {
         io.attachApp(server);
     }
 
+    io.engine.on("headers", (headers) => {
+        headers["x-host-id"] = process.env.HOSTNAME || "unknown";
+    });
+
     if (useRedis) {
         startCacheRefresher();
     }