diff --git a/upgrade.sh b/upgrade.sh
index 6095862..19b96f1 100644
--- a/upgrade.sh
+++ b/upgrade.sh
@@ -35,8 +35,9 @@ echo "Setting ownership and permissions for ca.key and ca.crt"
 chown hysteria:hysteria /etc/hysteria/ca.key /etc/hysteria/ca.crt
 chmod 640 /etc/hysteria/ca.key /etc/hysteria/ca.crt
 
-echo "Restarting hysteria-server.service"
+echo "Restarting hysteria and bot service"
 systemctl restart hysteria-server.service
+systemctl restart hysteria-bot.service
 
 echo "Setting execute permissions for user.sh and kick.sh"
 chmod +x /etc/hysteria/core/scripts/hysteria2/user.sh