diff --git a/tests/utils/test_chat_template_utils.py b/tests/utils/test_chat_template_utils.py
index 578c89f0e166c4..c1e0489350fe40 100644
--- a/tests/utils/test_chat_template_utils.py
+++ b/tests/utils/test_chat_template_utils.py
@@ -101,7 +101,7 @@ def fn(x: int = 42):
         self.assertEqual(schema, expected_schema)
 
     def test_nested_list(self):
-        def fn(x: List[List[Union[int, str]]]):
+        def fn(x: List[List[Union[str, int]]]):
             """
             Test function
 
@@ -109,8 +109,6 @@ def fn(x: List[List[Union[int, str]]]):
             """
             return x
 
-        self.maxDiff = None
-
         schema = get_json_schema(fn)
         expected_schema = {
             "name": "fn",
@@ -120,7 +118,7 @@ def fn(x: List[List[Union[int, str]]]):
                 "properties": {
                     "x": {
                         "type": "array",
-                        "items": {"type": "array", "items": {"type": ["integer", "string"]}},
+                        "items": {"type": "array", "items": {"type": ["string", "integer"]}},
                         "description": "The input",
                     }
                 },