diff --git a/src/backend/langflow/services/task/utils.py b/src/backend/langflow/services/task/utils.py
index 412b33ae2..e95fbdb33 100644
--- a/src/backend/langflow/services/task/utils.py
+++ b/src/backend/langflow/services/task/utils.py
@@ -1,6 +1,14 @@
-def get_celery_worker_status(app):
+import contextlib
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    with contextlib.suppress(ImportError):
+        from celery import Celery
+
+
+def get_celery_worker_status(app: "Celery"):
     i = app.control.inspect()
-    availability = i.ping()
+    availability = app.control.ping()
     stats = i.stats()
     registered_tasks = i.registered()
     active_tasks = i.active()
diff --git a/tests/conftest.py b/tests/conftest.py
index d6e25addf..c58f35cf5 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -74,14 +74,14 @@ class Config:
 @pytest.fixture(name="distributed_env")
 def setup_env(monkeypatch):
     monkeypatch.setenv("LANGFLOW_CACHE_TYPE", "redis")
-    monkeypatch.setenv("LANGFLOW_REDIS_HOST", "queue")
+    monkeypatch.setenv("LANGFLOW_REDIS_HOST", "result_backend")
     monkeypatch.setenv("LANGFLOW_REDIS_PORT", "6379")
     monkeypatch.setenv("LANGFLOW_REDIS_DB", "0")
     monkeypatch.setenv("LANGFLOW_REDIS_EXPIRE", "3600")
     monkeypatch.setenv("LANGFLOW_REDIS_PASSWORD", "")
     monkeypatch.setenv("FLOWER_UNAUTHENTICATED_API", "True")
-    monkeypatch.setenv("BROKER_URL", "redis://queue:6379/0")
-    monkeypatch.setenv("RESULT_BACKEND", "redis://queue:6379/0")
+    monkeypatch.setenv("BROKER_URL", "redis://result_backend:6379/0")
+    monkeypatch.setenv("RESULT_BACKEND", "redis://result_backend:6379/0")
     monkeypatch.setenv("C_FORCE_ROOT", "true")