diff --git a/src/typesense/types/document.py b/src/typesense/types/document.py
index 102ab02..3f0f9a8 100644
--- a/src/typesense/types/document.py
+++ b/src/typesense/types/document.py
@@ -300,6 +300,7 @@ class DocumentImportParametersReturnDocAndId(DocumentWriteParameters):
 
 
 DocumentImportParameters: typing.TypeAlias = typing.Union[
+    DocumentWriteParameters,
     DocumentImportParametersReturnId,
     DocumentImportParametersReturnDoc,
     DocumentImportParametersReturnDocAndId,