Color Coding fix