添加测试代码 尝试修复bug