代码重构后的BUG修复