Fix test comments