Fix color when done