fix tests
Merge pull request #10 from FooFoBerry/fix-tests

fix tests