Fix a missed test