修正单元测试
Merge pull request #89 from wangtuanjie/fix/unit_testing

修正单元测试