修改一下测试的代码