Fix a unit test