fix bug when press edit button