修复 #1, 并添加了一个简单的测试