fix tests
Merge pull request #2 from Parcela/fix-tests

fix tests