代码小修改，优化性能