修改部分代码逻辑