Fixed bug on undo_button