fix add/edit buttons