Fix broken unit test

I missed something in PR #9, fixing.