Fix edit buttons