Fixed test breakage from last commit - will still error due to celery bug