diff --git a/tests/test_agents_template.py b/tests/test_agents_template.py
index 750050c25..7aa8de176 100644
--- a/tests/test_agents_template.py
+++ b/tests/test_agents_template.py
@@ -82,6 +82,7 @@ def test_json_agent(client: TestClient):
         "type": "BaseLanguageModel",
         "list": False,
         "advanced": False,
+        "display_name": "LLM",
     }
 
 
@@ -120,6 +121,7 @@ def test_csv_agent(client: TestClient):
         "type": "BaseLanguageModel",
         "list": False,
         "advanced": False,
+        "display_name": "LLM",
     }
 
 
@@ -183,4 +185,5 @@ def test_initialize_agent(client: TestClient):
         "type": "BaseLanguageModel",
         "list": False,
         "advanced": False,
+        "display_name": "LLM",
     }
diff --git a/tests/test_chains_template.py b/tests/test_chains_template.py
index f1154a556..c958cf64d 100644
--- a/tests/test_chains_template.py
+++ b/tests/test_chains_template.py
@@ -326,6 +326,7 @@ def test_series_character_chain(client: TestClient):
 
     assert template["llm"] == {
         "required": True,
+        "display_name": "LLM",
         "placeholder": "",
         "show": True,
         "multiline": False,
@@ -388,6 +389,7 @@ def test_mid_journey_prompt_chain(client: TestClient):
 
     assert template["llm"] == {
         "required": True,
+        "display_name": "LLM",
         "placeholder": "",
         "show": True,
         "multiline": False,
@@ -427,6 +429,7 @@ def test_time_travel_guide_chain(client: TestClient):
     assert template["llm"] == {
         "required": True,
         "placeholder": "",
+        "display_name": "LLM",
         "show": True,
         "multiline": False,
         "password": False,