Fix undo/redo buttons