Problem Fix

Fix task_done problem.