diff --git a/tests/test_agents_template.py b/tests/test_agents_template.py
index 85f689eda..b12ad7dee 100644
--- a/tests/test_agents_template.py
+++ b/tests/test_agents_template.py
@@ -31,15 +31,15 @@ def test_zero_shot_agent(client: TestClient, logged_in_headers):
     }
 
     # Additional assertions for other template variables
-    assert template["callback_service"] == {
+    assert template["callback_manager"] == {
         "required": False,
         "dynamic": False,
         "placeholder": "",
         "show": False,
         "multiline": False,
         "password": False,
-        "name": "callback_service",
-        "type": "BaseCallbackService",
+        "name": "callback_manager",
+        "type": "BaseCallbackManager",
         "list": False,
         "advanced": False,
         "info": "",
diff --git a/tests/test_endpoints.py b/tests/test_endpoints.py
index da67cbb1a..cb3755db0 100644
--- a/tests/test_endpoints.py
+++ b/tests/test_endpoints.py
@@ -452,6 +452,7 @@ def test_basic_chat_with_two_session_ids_and_names(client, added_flow, created_a
 
 
 # Test function without loop
+@pytest.mark.async_test
 def test_async_task_processing(client, added_flow, created_api_key):
     headers = {"x-api-key": created_api_key.api_key}
     post_data = {"inputs": {"text": "Hi, My name is Gabriel"}}