Fix tests
Merge pull request #7 from aderuwe/patch-2

Fix tests