Fix tests
Merge pull request #4 from seblucas/master

Fix tests