Fix comment bug