Fix end comment bug