diff --git a/tradingagents/graph/portfolio_analysis.py b/tradingagents/graph/portfolio_analysis.py
index 52d1012c..96c51854 100644
--- a/tradingagents/graph/portfolio_analysis.py
+++ b/tradingagents/graph/portfolio_analysis.py
@@ -16,7 +16,7 @@ class PortfolioAnalyzer:
     owns the portfolio-level prompt, comparison logic, and logging.
     """
 
-    def __init__(self, deep_thinking_llm: ChatOpenAI):
+    def __init__(self, deep_thinking_llm: BaseChatModel):
         """Initialize with the deep thinking LLM for comparative analysis.
 
         Args: