fixing unit test
Merge pull request #3 from oh4real/master

fixing unit test