ENH: Fixed comment in code