Fix potential fail on comment coloring