fix tests
Merge pull request #16 from duselmann/cleanup

fix tests