修复部分代码