diff --git a/webui.sh b/webui.sh
index 8a3c6f12..246381fc 100755
--- a/webui.sh
+++ b/webui.sh
@@ -25,7 +25,7 @@ fi
 # Install directory without trailing slash
 if [[ -z "${install_dir}" ]]
 then
-    install_dir="$(dirname "$0")"
+    install_dir="$SCRIPT_DIR"
 fi
 
 # Name of the subdirectory (defaults to stable-diffusion-webui)