Fix color scheme