Fixes for new Celery api.