FIXES UNDO/REDO STATES