diff --git a/frontend/src/app/socketio/middleware.ts b/frontend/src/app/socketio/middleware.ts
index 1c2e11aff4..d4803810a4 100644
--- a/frontend/src/app/socketio/middleware.ts
+++ b/frontend/src/app/socketio/middleware.ts
@@ -22,9 +22,9 @@ import * as InvokeAI from '../invokeai';
  * some new action to handle whatever data was sent from the server.
  */
 export const socketioMiddleware = () => {
-  const { hostname, port } = new URL(window.location.href);
+  const { origin } = new URL(window.location.href);
 
-  const socketio = io(`http://${hostname}:${port}`, {
+  const socketio = io(origin, {
     timeout: 60000,
   });