1 files changed, 4 insertions, 4 deletions
diff --git a/node-repository/src/main/java/com/yahoo/vespa/hosted/provision/lb/LoadBalancerServiceMock.java b/node-repository/src/main/java/com/yahoo/vespa/hosted/provision/lb/LoadBalancerServiceMock.java
index 7da28bff930..df92a6ee44d 100644
--- a/node-repository/src/main/java/com/yahoo/vespa/hosted/provision/lb/LoadBalancerServiceMock.java
+++ b/node-repository/src/main/java/com/yahoo/vespa/hosted/provision/lb/LoadBalancerServiceMock.java
@@ -3,7 +3,6 @@ package com.yahoo.vespa.hosted.provision.lb;
 
 import ai.vespa.http.DomainName;
 import com.google.common.collect.ImmutableSet;
-import com.yahoo.config.provision.ApplicationId;
 import com.yahoo.config.provision.ClusterSpec;
 import com.yahoo.config.provision.NodeType;
 
@@ -60,14 +59,15 @@ public class LoadBalancerServiceMock implements LoadBalancerService {
                 Optional.of(new DnsZone("zone-id-1")),
                 Collections.singleton(4443),
                 ImmutableSet.of("10.2.3.0/24", "10.4.5.0/24"),
-                spec.reals());
+                spec.reals(),
+                spec.cloudAccount());
         instances.put(id, instance);
         return instance;
     }
 
     @Override
-    public void remove(ApplicationId application, ClusterSpec.Id cluster) {
-        instances.remove(new LoadBalancerId(application, cluster));
+    public void remove(LoadBalancer loadBalancer) {
+        instances.remove(loadBalancer.id());
     }
 
 }