Fix undo button not showing up