fix tests
Merge pull request #204 from maraoz/fix/broken/tests

Fix tests