修改一下测试代码