Fix edit button on wizard forms