diff --git a/samples/snippets/batch_process_documents_sample_v1beta3.py b/samples/snippets/batch_process_documents_sample_v1beta3.py
index 2936b3b3..6e22e0ea 100644
--- a/samples/snippets/batch_process_documents_sample_v1beta3.py
+++ b/samples/snippets/batch_process_documents_sample_v1beta3.py
@@ -110,7 +110,7 @@ def get_text(doc_element: dict, document: dict):
     for segment in doc_element.text_anchor.text_segments:
         start_index = (
             int(segment.start_index)
-            if "start_index" in doc_element.text_anchor.__dict__
+            if segment in doc_element.text_anchor.text_segments
             else 0
         )
         end_index = int(segment.end_index)
diff --git a/samples/snippets/process_document_sample_v1beta3.py b/samples/snippets/process_document_sample_v1beta3.py
index 330e8183..e666affa 100644
--- a/samples/snippets/process_document_sample_v1beta3.py
+++ b/samples/snippets/process_document_sample_v1beta3.py
@@ -77,7 +77,7 @@ def get_text(doc_element: dict, document: dict):
     for segment in doc_element.text_anchor.text_segments:
         start_index = (
             int(segment.start_index)
-            if segment.start_index in doc_element.text_anchor.text_segments
+            if segment in doc_element.text_anchor.text_segments
             else 0
         )
         end_index = int(segment.end_index)
diff --git a/samples/snippets/quickstart_sample_v1beta3.py b/samples/snippets/quickstart_sample_v1beta3.py
index c5cd34ae..7eb4a8ff 100644
--- a/samples/snippets/quickstart_sample_v1beta3.py
+++ b/samples/snippets/quickstart_sample_v1beta3.py
@@ -70,7 +70,7 @@ def get_text(doc_element: dict, document: dict):
     for segment in doc_element.text_anchor.text_segments:
         start_index = (
             int(segment.start_index)
-            if segment.start_index in doc_element.text_anchor.text_segments
+            if segment in doc_element.text_anchor.text_segments
             else 0
         )
         end_index = int(segment.end_index)