fix test
Merge pull request #6 from angryTit/feature-fix-test

fix test