Fixup broken unit test.