Fix failing test.
Merge pull request #15 from schwuk/fix_failing_test

Fix failing test.