修正重构的bug