fix small issue with displaying DONE button