fixing broken test
Merge pull request #4 from mrkrstphr/fix-tests

fixing broken test