Fix tests
Merge pull request #12 from akochurov/master

Fix tests