Fix a failing test case