LUN-507 : Fix undo button