Fix a broken test case