Fixed done button crash