修复重构时出现的一些BUG