fix tests
Merge pull request #21 from justplainboulevard/fix_tests

fix tests