fix some bug when click done button