Fix color_prompt