Fix test
Merge pull request #75 from vasichkin/master

Fix tests mock