Restart on upgrade