diff --git a/websocket_demo.html b/websocket_demo.html
index b57814e..59b3928 100644
--- a/websocket_demo.html
+++ b/websocket_demo.html
@@ -13,7 +13,7 @@
 <ul id='messageBox'>
 </ul>
 <script>
-    let ws = new WebSocket("ws://localhost:8000/ws");
+    let ws = new WebSocket("ws://" + location.host + "/ws");
     let history = [];
     let last_message_element = null;