fix tests
Merge pull request #3 from Mikxail/master

fix tests