ENH: Fixing silly bug