refactor bad code

重构一行代码。。。