Update to support celery 3.1.7+