diff --git a/tests/test_backend_msexcel.py b/tests/test_backend_msexcel.py
index e664ed3..f33dffa 100644
--- a/tests/test_backend_msexcel.py
+++ b/tests/test_backend_msexcel.py
@@ -3,7 +3,7 @@ import os
 from pathlib import Path
 
 from docling.datamodel.base_models import InputFormat
-from docling.datamodel.document import ConversionResult
+from docling.datamodel.document import ConversionResult, DoclingDocument
 from docling.document_converter import DocumentConverter
 
 GENERATE = False
diff --git a/tests/test_backend_msword.py b/tests/test_backend_msword.py
index 24db677..9edcb3e 100644
--- a/tests/test_backend_msword.py
+++ b/tests/test_backend_msword.py
@@ -6,6 +6,7 @@ from docling.backend.msword_backend import MsWordDocumentBackend
 from docling.datamodel.base_models import InputFormat
 from docling.datamodel.document import (
     ConversionResult,
+    DoclingDocument,
     InputDocument,
     SectionHeaderItem,
 )
diff --git a/tests/test_backend_pptx.py b/tests/test_backend_pptx.py
index 4c3872b..f4799a8 100644
--- a/tests/test_backend_pptx.py
+++ b/tests/test_backend_pptx.py
@@ -3,7 +3,7 @@ import os
 from pathlib import Path
 
 from docling.datamodel.base_models import InputFormat
-from docling.datamodel.document import ConversionResult
+from docling.datamodel.document import ConversionResult, DoclingDocument
 from docling.document_converter import DocumentConverter
 
 GENERATE = False