diff --git a/upgrade.sh b/upgrade.sh
index 8594860..c0f61d8 100644
--- a/upgrade.sh
+++ b/upgrade.sh
@@ -97,6 +97,7 @@ success "Python environment ready."
 
 # ========== Scheduler ==========
 info "Ensuring scheduler is set..."
+source "$HYSTERIA_INSTALL_DIR/core/scripts/scheduler.sh"
 if ! check_scheduler_service; then
     info "Installing Hysteria scheduler service..."
     setup_hysteria_scheduler