Fix tests
Merge pull request #1 from ASerputko/master

TEST: fixed broken tests