Fix model. Added set_celery_task_id