fixed problem with edit buttons