diff --git a/static/js/chat.js b/static/js/chat.js
index 7d35e8c..b967289 100644
--- a/static/js/chat.js
+++ b/static/js/chat.js
@@ -28,7 +28,11 @@ $(function(){
 function setupWS() {
     a = document.createElement("a");
     a.href = document.URL;
-    url = "ws://" + a.hostname + ":8000/ws/";
+    url = "ws://" + a.hostname;
+    if (a.port) {
+        url = url + ":" + a.port;
+    }
+    url = url + "/ws/";
     websocket = new WebSocket(url);
     websocket.onopen = function(evt) { onOpen(evt) };
     websocket.onclose = function(evt) { onClose(evt) };