diff --git a/lmm_tools/lmm/lmm.py b/lmm_tools/lmm/lmm.py
index e6f954a4..c5615114 100644
--- a/lmm_tools/lmm/lmm.py
+++ b/lmm_tools/lmm/lmm.py
@@ -1,7 +1,7 @@
 import base64
 from abc import ABC, abstractmethod
 from pathlib import Path
-from typing import Any, Dict, Optional, Union, cast
+from typing import Any, Dict, List, Optional, Union, cast
 
 
 def encode_image(image: Union[str, Path]) -> str:
@@ -36,7 +36,7 @@ def __init__(self, name: str):
         self.client = OpenAI()
 
     def generate(self, prompt: str, image: Optional[Union[str, Path]]) -> str:
-        message: list[Dict[str, Any]] = [
+        message: List[Dict[str, Any]] = [
             {
                 "role": "user",
                 "content": [