Fix undo for color change