fix up a unit test