diff --git a/run b/run
index 9ea6996..f43f10c 100755
--- a/run
+++ b/run
@@ -1,6 +1,7 @@
-#!/usr/bin/env python
+#!/usr/bin/env sh
 
-import websockify
+BASE_DIR="$(dirname "$(realpath "$0")")"
 
-if __name__ == '__main__':
-	websockify.websocketproxy.websockify_init()
+cd "$BASE_DIR"
+
+python -m websockify $@
diff --git a/websockify/__main__.py b/websockify/__main__.py
new file mode 100644
index 0000000..8378d46
--- /dev/null
+++ b/websockify/__main__.py
@@ -0,0 +1,4 @@
+import websockify
+
+if __name__ == '__main__':
+    websockify.websocketproxy.websockify_init()