fix tests
Merge pull request #79 from dereuromark/master-tests

fix tests