优化部分逻辑

优化部分逻辑