fix tests
Merge pull request #20 from louridas/master

fix tests