Minor amendment to annotation in test