minor bug fix , added comment