Fix celery tasks