diff --git a/app/services/improved_search_service.py b/app/services/improved_search_service.py
index c3acfe4..3985137 100644
--- a/app/services/improved_search_service.py
+++ b/app/services/improved_search_service.py
@@ -12,6 +12,7 @@ from urllib.parse import quote_plus, urljoin, urlparse
 
 from bs4 import BeautifulSoup
 from playwright.async_api import Browser, BrowserContext, Page, async_playwright
+from sqlalchemy.orm import Session
 
 
 from app.core.search_config import SITE_CONFIGS, BROWSERLESS_URL