Fix for celery import error