Fix a unit test.