Fix tests celery