Fix preview edit button