AI undo/redo bug fix