修正测试代码