1 files changed, 4 insertions, 1 deletions
diff --git a/src/server-component.mjs b/src/server-component.mjs
index ab08801..0700939 100644
--- a/src/server-component.mjs
+++ b/src/server-component.mjs
@@ -60,9 +60,12 @@ export class LustreServerComponent extends HTMLElement {
         } else if (prev !== next) {
           const id = this.getAttribute("id");
           const route = next + (id ? `?id=${id}` : "");
+          const protocol = window.location.protocol === "https:" ? "wss" : "ws";
 
           this.#socket?.close();
-          this.#socket = new WebSocket(`ws://${window.location.host}${route}`);
+          this.#socket = new WebSocket(
+            `${protocol}://${window.location.host}${route}`,
+          );
           this.#socket.addEventListener("message", (message) =>
             this.messageReceivedCallback(message),
           );