Ensure compatibility with celery