修复单元测试bug