diff --git a/demo/components/WebSocketExample.tsx b/demo/components/WebSocketExample.tsx
index b232bfe..187f8f1 100644
--- a/demo/components/WebSocketExample.tsx
+++ b/demo/components/WebSocketExample.tsx
@@ -12,7 +12,7 @@ export interface ServerWSResponse {
 }
 
 export default function WebSocketExample({port = 8080}:{port?:number}){
-  const server_ws_url  = process.env.NODE_ENV === "production" ? `ws://localhost:${port}` : "ws://new-media.fly.dev"
+  const server_ws_url  = process.env.NODE_ENV != "production" ? `wss://localhost:${port}` : "ws://new-media.fly.dev"
   const [imgMetadata, setImageMetadata] = useState<ServerWSResponse>();
   const [socket, setSocket] = useState<WebSocket>()