[tests] Fix tests
Merge pull request #12 from martyone/fix-tests

[tests] Fix tests