修复单元测试的一些问题