fix in the undo / redo