diff --git a/src/vectara/managers/document.py b/src/vectara/managers/document.py
index fab05f3..1336ca5 100644
--- a/src/vectara/managers/document.py
+++ b/src/vectara/managers/document.py
@@ -1,5 +1,5 @@
 from vectara.documents.client import DocumentsClient
-from vectara.types import Document, CoreDocument, StructuredDocument
+from vectara.types import Document, CreateDocumentRequest_Core, CreateDocumentRequest_Structured
 from vectara.utils.hash import calculate_sha256
 from typing import Optional, Dict, Any, Union, Tuple
 from enum import Enum
@@ -90,7 +90,7 @@ def check_same(self, corpus_key: str, doc_id: str, content: bytes, metadata: Opt
             return True, True
 
 
-    def index_doc(self, corpus_key: str, doc: Union[CoreDocument, StructuredDocument]) -> DocOpEnum:
+    def index_doc(self, corpus_key: str, doc: Union[CreateDocumentRequest_Core, CreateDocumentRequest_Structured]) -> DocOpEnum:
 
         # Remove the sha256 hash from metadata.
         if doc.metadata and HASH_FIELD in doc.metadata: