Color scheme fix