修正部分代码。

修正部分代码。