Fix a bug in color scheme setup