进一步修正代码结构