Fix an issue with a unit test.