diff --git a/functions/self_update.sh b/functions/self_update.sh
index 586cadc9..5c2178d0 100644
--- a/functions/self_update.sh
+++ b/functions/self_update.sh
@@ -4,8 +4,7 @@ args=("$@")
 self_update() {
 git fetch &> /dev/null 
 echo "🅂 🄴 🄻 🄵   🅄 🄿 🄳 🄰 🅃 🄴"
-# TODO: change beta to main once testing is complete 
-if  git diff --name-only origin/beta | grep -qs ".sh" ; then
+if  git diff --name-only origin/main | grep -qs ".sh" ; then
     echo "Found a new version of HeavyScript, updating myself..."
     git reset --hard -q
     git pull --force -q
@@ -25,4 +24,4 @@ else
     echo -e "HeavyScript is already the latest version\n"
 fi
 }
-export -f self_update
\ No newline at end of file
+export -f self_update