diff --git a/src/main.js b/src/main.js
index 45bedcb..cbac006 100644
--- a/src/main.js
+++ b/src/main.js
@@ -15,7 +15,7 @@ if (qs.get(`ws-uri`)) {
 
 // http://localhost:8080/?ws-uri=localhost:3000
 
-while (ws_uri.length == 0 || !ws_uri.match(/^(wss?|https?):\/\//)) {
+while ((!ws_uri || ws_uri.length == 0) || !ws_uri.match(/^(wss?|https?):\/\//)) {
 	ws_uri = prompt("Input a websocket URI to connect to");
 }