fix bug
优化代码