diff --git a/src/trading/journal.py b/src/trading/journal.py
index de6abca..641f021 100644
--- a/src/trading/journal.py
+++ b/src/trading/journal.py
@@ -6,7 +6,7 @@ from zoneinfo import ZoneInfo
 import yfinance as yf
 from db.db_connection import create_client
 from trading.position_calculator import PositionCalculator
-from utils.data_utils import get_user_input
+from utils.data_utils import get_user_input, get_current_prices
 
 def handle_sell_order(ticker: str, shares_to_sell: int, exit_price: float, exit_date: datetime,
                      order_type: str, followed_rules: bool, exit_reason: str, notes: Optional[str] = None) -> bool: