ENH: Minor change