Fix unit test from #205