diff --git a/langchain4j-neo4j/src/test/java/dev/langchain4j/store/embedding/neo4j/Neo4jEmbeddingStoreTest.java b/langchain4j-neo4j/src/test/java/dev/langchain4j/store/embedding/neo4j/Neo4jEmbeddingStoreTest.java
index 607427062c..dc666f5770 100644
--- a/langchain4j-neo4j/src/test/java/dev/langchain4j/store/embedding/neo4j/Neo4jEmbeddingStoreTest.java
+++ b/langchain4j-neo4j/src/test/java/dev/langchain4j/store/embedding/neo4j/Neo4jEmbeddingStoreTest.java
@@ -255,8 +255,13 @@ void should_add_multiple_embeddings() {
 
         checkEntitiesCreated(relevant.size(), 
                 iterator -> {
-            checkDefaultProps(firstEmbedding, firstMatch, iterator.next());
-            checkDefaultProps(secondEmbedding, secondMatch, iterator.next());
+                    iterator.forEachRemaining(node -> {
+                        if (node.get(DEFAULT_ID_PROP).asString().equals(firstMatch.embeddingId())) {
+                            checkDefaultProps(firstEmbedding, firstMatch, node);
+                        } else {
+                            checkDefaultProps(secondEmbedding, secondMatch, node);
+                        }
+                    });
         });
     }
 
@@ -289,12 +294,17 @@ void should_add_multiple_embeddings_with_segments() {
         assertThat(secondMatch.embedding()).isEqualTo(secondEmbedding);
         assertThat(secondMatch.embedded()).isEqualTo(secondSegment);
 
-        checkEntitiesCreated(relevant.size(), 
+        checkEntitiesCreated(relevant.size(),
                 iterator -> {
-            List<String> otherProps = Collections.singletonList(DEFAULT_TEXT_PROP);
-            checkDefaultProps(firstEmbedding, firstMatch, iterator.next(), otherProps);
-            checkDefaultProps(secondEmbedding, secondMatch, iterator.next(), otherProps);
-        });
+                    List<String> otherProps = Collections.singletonList(DEFAULT_TEXT_PROP);
+                    iterator.forEachRemaining(node -> {
+                        if (node.get(DEFAULT_ID_PROP).asString().equals(firstMatch.embeddingId())) {
+                            checkDefaultProps(firstEmbedding, firstMatch, node, otherProps);
+                        } else {
+                            checkDefaultProps(secondEmbedding, secondMatch, node, otherProps);
+                        }
+                    });
+                });
     }
 
     @Test
@@ -345,8 +355,13 @@ void should_find_with_min_score() {
 
         checkEntitiesCreated(relevant.size(),
                 iterator -> {
-            checkDefaultProps(firstEmbedding, firstMatch, iterator.next());
-            checkDefaultProps(secondEmbedding, secondMatch, iterator.next());
+                    iterator.forEachRemaining(node -> {
+                        if (node.get(DEFAULT_ID_PROP).asString().equals(firstMatch.embeddingId())) {
+                            checkDefaultProps(firstEmbedding, firstMatch, node);
+                        } else {
+                            checkDefaultProps(secondEmbedding, secondMatch, node);
+                        }
+                    });
         });
     }