Fix annotation in test