fixed edit button bug