fix bug for done
fix bug for done