diff --git a/deepsearch/documents/core/export.py b/deepsearch/documents/core/export.py
index b6a73a91..77aaa869 100644
--- a/deepsearch/documents/core/export.py
+++ b/deepsearch/documents/core/export.py
@@ -3,7 +3,7 @@
 import warnings
 from typing import Any, Dict
 
-from docling_core.types import Document as DsDocument
+from docling_core.types.legacy_doc.document import ExportedCCSDocument as DsDocument
 
 _log = logging.getLogger(__name__)
 
diff --git a/deepsearch/documents/core/utils.py b/deepsearch/documents/core/utils.py
index 1d80cb81..501bc7a6 100644
--- a/deepsearch/documents/core/utils.py
+++ b/deepsearch/documents/core/utils.py
@@ -9,7 +9,7 @@
 from typing import Any, Dict, Iterator, List, Optional
 
 import requests
-from docling_core.types import Document as DsDocument
+from docling_core.types.legacy_doc.document import ExportedCCSDocument as DsDocument
 from pydantic import BaseModel
 from tqdm import tqdm