Fixed the missing button after editing issue.