Fix color scheme toggleing