Fix comment coloring