Fix a unit test failure