Fixes undo! and redo! methods