Better integration with celery