Fix an potential issue that coloring doesn't update