small fix to edit button