fixed test
Merge pull request #20 from ankri/testfix

Fixed bug in tests