修复测试代码