Fix a minor test