diff --git a/providers/pinecone/tests/system/pinecone/example_pinecone_cohere.py b/providers/pinecone/tests/system/pinecone/example_pinecone_cohere.py
index 496c0a40154bf..2e74fe870a07d 100644
--- a/providers/pinecone/tests/system/pinecone/example_pinecone_cohere.py
+++ b/providers/pinecone/tests/system/pinecone/example_pinecone_cohere.py
@@ -54,7 +54,7 @@ def create_index():
     @task
     def transform_output(embedding_output) -> list[dict]:
         # Convert each embedding to a map with an ID and the embedding vector
-        return [dict(id=str(i), values=embedding) for i, embedding in enumerate(embedding_output.float_)]
+        return [dict(id=str(i), values=embedding) for i, embedding in enumerate(embedding_output)]
 
     transformed_output = transform_output(embed_task.output)