Fix missing end comment