Fixed test failure
Merge pull request #15 from passy/test-fix

Fixed test failure