fix unit tests
Merge pull request #28 from newzly/tmp

fix unit tests