修正单元测试问题