diff --git a/scripts/on_env_start.bat b/scripts/on_env_start.bat
index 76fdb712..381201ce 100644
--- a/scripts/on_env_start.bat
+++ b/scripts/on_env_start.bat
@@ -25,6 +25,7 @@
     @cd sd-ui-files
 
     @call git reset --hard
+    @call git checkout main
     @call git pull
 
     @cd ..
diff --git a/scripts/on_env_start.sh b/scripts/on_env_start.sh
index 044f9124..bb24a668 100755
--- a/scripts/on_env_start.sh
+++ b/scripts/on_env_start.sh
@@ -8,6 +8,7 @@ if [ -f "scripts/install_status.txt" ] && [ `grep -c sd_ui_git_cloned scripts/in
     cd sd-ui-files
 
     git reset --hard
+    git checkout main
     git pull
 
     cd ..