diff --git a/lib/functions.zsh b/lib/functions.zsh
index 0d632a268..5f87f27c1 100644
--- a/lib/functions.zsh
+++ b/lib/functions.zsh
@@ -8,6 +8,10 @@ function uninstall_oh_my_zsh() {
 
 function upgrade_oh_my_zsh() {
   env ZSH=$ZSH /bin/sh $ZSH/tools/upgrade.sh
+  if [ -x $ZSH/custom/tools/upgrade.sh ]
+  then
+    env ZSH=$ZSH /bin/sh $ZSH/custom/tools/upgrade.sh
+  fi
 }
 
 function take() {
diff --git a/tools/check_for_upgrade.sh b/tools/check_for_upgrade.sh
index 5afd83c1e..188b09bfb 100644
--- a/tools/check_for_upgrade.sh
+++ b/tools/check_for_upgrade.sh
@@ -14,6 +14,10 @@ function _upgrade_zsh() {
   env ZSH=$ZSH /bin/sh $ZSH/tools/upgrade.sh
   # update the zsh file
   _update_zsh_update
+  if [ -x $ZSH/custom/tools/upgrade.sh ]
+  then
+    env ZSH=$ZSH /bin/sh $ZSH/custom/tools/upgrade.sh
+  fi
 }
 
 epoch_target=$UPDATE_ZSH_DAYS