fix unit test
Merge pull request #5 from grogy/fix-test

fix unit test