Fix Unit Test
Merge pull request #54 from lingoer/master

Fix Unit Test