重构，修复bug