Updated tests to use new celery task