Fixed a error on click done