diff --git a/app/llm.py b/app/llm.py
index 4b68e7d..87c59f6 100644
--- a/app/llm.py
+++ b/app/llm.py
@@ -10,7 +10,7 @@ from openai import (
     OpenAIError,
     RateLimitError,
 )
-from openai.types.chat.chat_completion_message import ChatCompletionMessage
+from openai.types.chat import ChatCompletion, ChatCompletionMessage
 from tenacity import (
     retry,
     retry_if_exception_type,