Make celery tests work