Minor edit in test