Fix a broken test