Fix tests
Merge pull request #11 from dennishendriksen/master

Fix tests