Fix broken tests
Merge pull request #29 from camelmasa/hotfix/broken-test

Fix broken tests