diff --git a/tools/check_for_upgrade.sh b/tools/check_for_upgrade.sh
index 4643739ca..a7b9db5ab 100644
--- a/tools/check_for_upgrade.sh
+++ b/tools/check_for_upgrade.sh
@@ -25,10 +25,9 @@ then
     if [ "$line" = Y ] || [ "$line" = y ]
     then
       /bin/sh $ZSH/tools/upgrade.sh
+
+      # update the zsh file
+      _update_zsh_update
     fi
   fi
 fi
-
-# update the zsh file
-_update_zsh_update
-