fix tests
Merge pull request #14 from mpuncel/mpuncel/fix-tests

fix tests