Bug fix
added comment