Fix test case
Merge pull request #13 from kittylyst/master

Fix test case