Fixed tests
Merge pull request #23 from enumag/patch-tests

Fixed tests