Fix coloring in generic tokens