Fixed add comment bug