added fix for last test
Merge pull request #13 from olegshegula/olegsbranch

added fix for last test