Fix a broken unit test