fix tests
Merge pull request #1 from edjafarov/master

fix tests:)