diff --git a/tests/test_backend_pptx.py b/tests/test_backend_pptx.py
index 4c3872be..f4799a83 100644
--- a/tests/test_backend_pptx.py
+++ b/tests/test_backend_pptx.py
@@ -3,7 +3,7 @@
 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