fix tests
Merge pull request #8 from Bronsa/master

fix tests