修正逻辑，使得更简单