diff --git a/flaskserver.py b/flaskserver.py
index af0b14b..750039b 100755
--- a/flaskserver.py
+++ b/flaskserver.py
@@ -48,10 +48,12 @@ print(args)
 
 @app.route('/')
 def index():
-    if request.args.get('tunneled'):
-        wsaddr = f'127.0.0.1:{args.wsport}'
-    else:
-        wsaddr = f'{args.hostname}:{args.wsport}'
+    wsaddr = request.args.get('wsaddr')
+    if not wsaddr:
+        if request.args.get('tunneled'):
+            wsaddr = f'127.0.0.1:{args.wsport}'
+        else:
+            wsaddr = f'{args.hostname}:{args.wsport}'
     print(wsaddr, args.port, args.html, args.args)
     return render_template(f'{args.html}', wsaddr=wsaddr, **args.args)