fixed done-button bug