diff --git a/src/funcchain/model/abilities.py b/src/funcchain/model/abilities.py
index a796344..82c3b56 100644
--- a/src/funcchain/model/abilities.py
+++ b/src/funcchain/model/abilities.py
@@ -29,7 +29,7 @@
 
 
 def gather_llm_type(llm: BaseChatModel, func_check: bool = True) -> str:
-    from langchain_openai.chat_models import ChatOpenAI
+    from .patches import ChatOpenAI
 
     if not isinstance(llm, BaseChatModel):
         return "base_model"