diff --git a/test/agentchat/test_structured_output.py b/test/agentchat/test_structured_output.py
index 24d90d3c6a..614cb8f604 100644
--- a/test/agentchat/test_structured_output.py
+++ b/test/agentchat/test_structured_output.py
@@ -7,6 +7,7 @@
 
 import os
 import sys
+from typing import List
 from unittest.mock import MagicMock
 
 import pytest
@@ -70,7 +71,7 @@ class Step(BaseModel):
 
 
 class MathReasoning(BaseModel):
-    steps: list[Step]
+    steps: List[Step]
     final_answer: str
 
     def format(self) -> str: