diff --git a/update b/update
index c766aaac..c92b1450 100755
--- a/update
+++ b/update
@@ -159,9 +159,23 @@ fi
 
 cd /home/pi/pi-power-button
 
-git checkout reboot-mode-change-beta
+git checkout reboot-mode-change-beta > .updated_b
 
-git pull --no-rebase > .updated_p
+  grep 'error' /home/pi/pi-power-button/.updated_b
+  if [[ $(grep 'error' /home/pi/pi-power-button/.updated_b) ]]; then
+  
+    echo "pi-power-button switching to beta branch"
+
+    git pull
+
+    git checkout reboot-mode-change-beta
+    
+    script/install
+
+    FLAG=1
+  fi
+
+  git pull --no-rebase > .updated_p
 
   grep 'changed' /home/pi/pi-power-button/.updated_p
   if [[ $(grep 'changed' /home/pi/pi-power-button/.updated_p) ]]; then