fix a unit test