修改部分测试代码