diff --git a/scripts/on_sd_start.sh b/scripts/on_sd_start.sh
index 204a003c..57bf15b7 100755
--- a/scripts/on_sd_start.sh
+++ b/scripts/on_sd_start.sh
@@ -64,5 +64,6 @@ else
     # Download the models
     script_dir="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
     ui_absolute_path=$(readlink -f "$script_dir/../ui")
-    PYTHONPATH="$ui_absolute_path" SD_UI_DIR="$ui_absolute_path" python -c "from easydiffusion.model_manager import init; init()"
+    export SD_UI_DIR="$ui_absolute_path"
+    PYTHONPATH="$ui_absolute_path" python -c "from easydiffusion.model_manager import init; init()"
 fi