diff --git a/tradingagents/graph/portfolio_analysis.py b/tradingagents/graph/portfolio_analysis.py
index 728985d8..6409258b 100644
--- a/tradingagents/graph/portfolio_analysis.py
+++ b/tradingagents/graph/portfolio_analysis.py
@@ -1,6 +1,7 @@
 # TradingAgents/graph/portfolio_analysis.py
 
 import json
+import re
 import traceback
 from pathlib import Path
 from typing import Any, Callable, Dict, List, Tuple