fixing undo/redo again