Fix done button status issue