diff --git a/trustgraph-flow/trustgraph/embeddings/document_embeddings/embeddings.py b/trustgraph-flow/trustgraph/embeddings/document_embeddings/embeddings.py
index eb21d418..f038a9b5 100755
--- a/trustgraph-flow/trustgraph/embeddings/document_embeddings/embeddings.py
+++ b/trustgraph-flow/trustgraph/embeddings/document_embeddings/embeddings.py
@@ -67,8 +67,7 @@ class Processor(FlowProcessor):
             )
 
             # vectors[0] is the vector set for the first (only) text
-            # vectors[0][0] is the first vector in that set
-            vectors = resp.vectors[0][0] if resp.vectors else []
+            vectors = resp.vectors[0] if resp.vectors else []
 
             embeds = [
                 ChunkEmbeddings(
diff --git a/trustgraph-flow/trustgraph/embeddings/graph_embeddings/embeddings.py b/trustgraph-flow/trustgraph/embeddings/graph_embeddings/embeddings.py
index c54e719d..e83d608b 100755
--- a/trustgraph-flow/trustgraph/embeddings/graph_embeddings/embeddings.py
+++ b/trustgraph-flow/trustgraph/embeddings/graph_embeddings/embeddings.py
@@ -72,7 +72,7 @@ class Processor(FlowProcessor):
             entities = [
                 EntityEmbeddings(
                     entity=entity.entity,
-                    vectors=vectors[0],  # First vector from the set
+                    vectors=vectors,  # Vector set for this entity
                     chunk_id=entity.chunk_id,  # Provenance: source chunk
                 )
                 for entity, vectors in zip(v.entities, all_vectors)
diff --git a/trustgraph-flow/trustgraph/embeddings/row_embeddings/embeddings.py b/trustgraph-flow/trustgraph/embeddings/row_embeddings/embeddings.py
index c1d04302..f81e4374 100644
--- a/trustgraph-flow/trustgraph/embeddings/row_embeddings/embeddings.py
+++ b/trustgraph-flow/trustgraph/embeddings/row_embeddings/embeddings.py
@@ -216,7 +216,7 @@ class Processor(CollectionConfigHandler, FlowProcessor):
                         index_name=index_name,
                         index_value=index_value,
                         text=text,
-                        vectors=vectors[0]  # First vector from the set
+                        vectors=vectors  # Vector set for this text
                     )
                 )