fix broken undo button