Fix failing test.
Merge pull request #42 from frankban/fix-test-error

Fix failing test.