修正一处导致Restart的bug，简化逻辑