Fixing the tests
Merge pull request #39 from OlegIlyenko/master

Fixing the tests