Fix code coloring