Change to using a mock instead of the celery task