diff --git a/algo/src/test/java/org/neo4j/gds/similarity/filteredknn/FilteredKnnIdMappingTest.java b/algo/src/test/java/org/neo4j/gds/similarity/filteredknn/FilteredKnnIdMappingTest.java
index 2a43c3fc34..8e44f1402e 100644
--- a/algo/src/test/java/org/neo4j/gds/similarity/filteredknn/FilteredKnnIdMappingTest.java
+++ b/algo/src/test/java/org/neo4j/gds/similarity/filteredknn/FilteredKnnIdMappingTest.java
@@ -74,7 +74,8 @@ void shouldIdMapTheSourceNodeFilter() {
         var sourceNodesInResult = result
             .similarityResultStream()
             .map(res -> res.node1)
+            .map(graph::toOriginalNodeId)
             .collect(Collectors.<Long>toSet());
-        assertThat(sourceNodesInResult).containsExactly(0L);
+        assertThat(sourceNodesInResult).containsExactly(lowestNeoId);
     }
 }