add comment bug fix