Fix stop_task