Fixed bug for edit/delete buttons