fix tests
Merge pull request #8 from smn/feature/fix-tests

Feature/fix tests