Fix tests

1 files changed, 3 insertions, 2 deletions

diff --git a/node-repository/src/main/java/com/yahoo/vespa/hosted/provision/persistence/CuratorDatabaseClient.java b/node-repository/src/main/java/com/yahoo/vespa/hosted/provision/persistence/CuratorDatabaseClient.java
index 379bb2566df..a9abc352d8c 100644
--- a/node-repository/src/main/java/com/yahoo/vespa/hosted/provision/persistence/CuratorDatabaseClient.java
+++ b/node-repository/src/main/java/com/yahoo/vespa/hosted/provision/persistence/CuratorDatabaseClient.java
@@ -43,8 +43,8 @@ import java.util.logging.Logger;
 import java.util.stream.Collectors;
 import java.util.stream.IntStream;
 
+import static com.yahoo.stream.CustomCollectors.toLinkedMap;
 import static java.util.stream.Collectors.collectingAndThen;
-import static java.util.stream.Collectors.toMap;
 
 /**
  * Client which reads and writes nodes to a curator database.
@@ -453,7 +453,8 @@ public class CuratorDatabaseClient {
                                           .map(this::readLoadBalancer)
                                           .filter(Optional::isPresent)
                                           .map(Optional::get)
-                                          .collect(collectingAndThen(toMap(LoadBalancer::id, Function.identity()),
+                                          .collect(collectingAndThen(toLinkedMap(LoadBalancer::id,
+                                                                                 Function.identity()),
                                                                      Collections::unmodifiableMap));
     }