diff --git a/tests/test_backend_msword.py b/tests/test_backend_msword.py
index 24db6775..9edcb3e6 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,
 )