PR #200 - Fix broken test

Sorry, I've broke the tests with #200