From 964d752e119fd5944ad98e8a404fe4df65e70c84 Mon Sep 17 00:00:00 2001
From: cmdr2 <secondary.cmdr2@gmail.com>
Date: Thu, 8 Sep 2022 23:28:22 +0530
Subject: [PATCH] Fix

---
 scripts/on_env_start.bat | 1 +
 scripts/on_env_start.sh  | 1 +
 2 files changed, 2 insertions(+)

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 ..