diff --git a/tests/test_llms_template.py b/tests/test_llms_template.py
index 14e151479..3da176192 100644
--- a/tests/test_llms_template.py
+++ b/tests/test_llms_template.py
@@ -2,15 +2,6 @@ from fastapi.testclient import TestClient
 from langflow.services.utils import get_settings_manager
 
 
-def test_llms_settings(client: TestClient, logged_in_headers):
-    settings_manager = get_settings_manager()
-    response = client.get("api/v1/all", headers=logged_in_headers)
-    assert response.status_code == 200
-    json_response = response.json()
-    llms = json_response["llms"]
-    assert set(llms.keys()) == set(settings_manager.settings.LLMS)
-
-
 # def test_hugging_face_hub(client: TestClient):
 #     response = client.get("api/v1/all")
 #     assert response.status_code == 200
@@ -85,7 +76,7 @@ def test_llms_settings(client: TestClient, logged_in_headers):
 #         "multiline": False,
 #         "password": False,
 #         "name": "model_kwargs",
-#         "type": "code",
+#         "type": "dict" ,
 #         "list": False,
 #         "advanced": True,
 #     }
@@ -279,7 +270,7 @@ def test_openai(client: TestClient, logged_in_headers):
         "multiline": False,
         "password": False,
         "name": "model_kwargs",
-        "type": "code",
+        "type": "dict",
         "list": False,
         "advanced": True,
         "info": "",
@@ -334,7 +325,7 @@ def test_openai(client: TestClient, logged_in_headers):
         "multiline": False,
         "password": False,
         "name": "logit_bias",
-        "type": "code",
+        "type": "dict",
         "list": False,
         "advanced": False,
         "info": "",
@@ -451,7 +442,7 @@ def test_chat_open_ai(client: TestClient, logged_in_headers):
         "multiline": False,
         "password": False,
         "name": "model_kwargs",
-        "type": "code",
+        "type": "dict",
         "list": False,
         "advanced": True,
         "info": "",
diff --git a/tests/test_prompts_template.py b/tests/test_prompts_template.py
index 676448f73..434691038 100644
--- a/tests/test_prompts_template.py
+++ b/tests/test_prompts_template.py
@@ -55,7 +55,7 @@ def test_prompt_template(client: TestClient, logged_in_headers):
         "multiline": False,
         "password": False,
         "name": "partial_variables",
-        "type": "code",
+        "type": "dict",
         "list": False,
         "advanced": False,
         "info": "",
diff --git a/tests/test_template.py b/tests/test_template.py
index 4be6dff06..81f2a6020 100644
--- a/tests/test_template.py
+++ b/tests/test_template.py
@@ -135,7 +135,7 @@ def test_format_dict():
     }
     expected_output = {
         "field1": {
-            "type": "code",  # Mapping type is replaced with dict which is replaced with code
+            "type": "dict[str, int]",  # Mapping type is replaced with dict which is replaced with code
             "required": False,
             "list": False,
             "show": False,
@@ -249,7 +249,7 @@ def test_format_dict():
     }
     expected_output = {
         "field1": {
-            "type": "code",
+            "type": "Dict[str, int]",
             "required": False,
             "list": False,
             "show": False,