Fixing undo.