Fixed done button bug