Fix color reset again