fix unit test
Merge pull request #4 from jesuspg/master

fix unit test