Fix celery refs