修复单元测试失败