ENH: Minor fixes