修正部分代码