Fix celery tests