diff --git a/kernel/setup.sh b/kernel/setup.sh
index 856d1cd6..98c9564a 100755
--- a/kernel/setup.sh
+++ b/kernel/setup.sh
@@ -17,7 +17,11 @@ fi
 
 test -d "$GKI_ROOT/KernelSU" || git clone https://github.com/tiann/KernelSU
 cd "$GKI_ROOT/KernelSU"
-git stash && git pull
+git stash
+if [ "$(git status | grep -Po 'v\d+(\.\d+)*' | head -n1)" ]; then
+     git checkout main
+fi
+git pull
 if [ -z "${1-}" ]; then
     git checkout "$(git describe --abbrev=0 --tags)"
 else