fix tests
Merge pull request #6 from Parcela/fix-test

fix tests