Minor fix to edit button placement.