修正一个错误
整理代码