diff --git a/lib/functions.zsh b/lib/functions.zsh
index f9d4a9717..ef7a5cf97 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 881bc6c49..e17b101d7 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