优化代码

修复了一个逻辑错误