diff --git a/application/api/user/routes.py b/application/api/user/routes.py
index fba818c..56b0201 100644
--- a/application/api/user/routes.py
+++ b/application/api/user/routes.py
@@ -6,7 +6,6 @@ from pymongo import MongoClient
 from bson.objectid import ObjectId
 from werkzeug.utils import secure_filename
 import http.client
-from celery.result import AsyncResult
 
 from application.api.user.tasks import ingest
 
@@ -142,7 +141,8 @@ def upload_file():
 def task_status():
     """Get celery job status."""
     task_id = request.args.get("task_id")
-    task = AsyncResult(task_id, backend=settings.CELERY_RESULT_BACKEND)
+    from application.celery import celery
+    task = celery.AsyncResult(task_id)
     task_meta = task.info
     return {"status": task.status, "result": task_meta}
 
diff --git a/application/celery.py b/application/celery.py
index 3b4819e..c19c2e7 100644
--- a/application/celery.py
+++ b/application/celery.py
@@ -2,7 +2,7 @@ from celery import Celery
 from application.core.settings import settings
 
 def make_celery(app_name=__name__):
-    celery = Celery(app_name, broker=settings.CELERY_BROKER_URL)
+    celery = Celery(app_name, broker=settings.CELERY_BROKER_URL, backend=settings.CELERY_RESULT_BACKEND)
     celery.conf.update(settings)
     return celery
 
diff --git a/setup.sh b/setup.sh
index 43e708d..f9d3b87 100755
--- a/setup.sh
+++ b/setup.sh
@@ -35,6 +35,8 @@ download_locally() {
     #pip install -r application/requirements.txt
     #pip install llama-cpp-python
     #pip install sentence-transformers
+    export LLM_NAME=llama.cpp
+    export EMBEDDINGS_NAME=huggingface_sentence-transformers/all-mpnet-base-v2
     export FLASK_APP=application/app.py
     export FLASK_DEBUG=true
     export CELERY_BROKER_URL=redis://localhost:6379/0
@@ -42,7 +44,7 @@ download_locally() {
     echo "The application is now running on http://localhost:5173"
     echo "You can stop the application by running the following command:"
     echo "Ctrl + C and then"
-    echo "Then pkill -f "flask run" and then"
+    echo "Then pkill -f 'flask run' and then"
     echo "docker-compose down"
     flask run --host=0.0.0.0 --port=7091 &
     celery -A application.app.celery worker -l INFO