Fixed second unit test
Merge pull request #9 from wesleycho/master

Fixed second unit test