Follow API change

1 files changed, 8 insertions, 3 deletions

diff --git a/configserver/src/test/java/com/yahoo/vespa/config/server/MockProvisioner.java b/configserver/src/test/java/com/yahoo/vespa/config/server/MockProvisioner.java
index 2c9b3c20d4c..797f895737b 100644
--- a/configserver/src/test/java/com/yahoo/vespa/config/server/MockProvisioner.java
+++ b/configserver/src/test/java/com/yahoo/vespa/config/server/MockProvisioner.java
@@ -74,14 +74,19 @@ public class MockProvisioner implements Provisioner {
     }
 
     @Override
+    public void remove(ApplicationTransaction transaction) {
+        removed = true;
+        lastApplicationId = transaction.application();
+    }
+
+    @Override
     public void remove(NestedTransaction transaction, ApplicationId application) {
-        remove(transaction, lock(application));
+        remove(new ApplicationTransaction(lock(application), transaction));
     }
 
     @Override
     public void remove(NestedTransaction transaction, ProvisionLock lock) {
-        removed = true;
-        lastApplicationId = lock.application();
+        remove(new ApplicationTransaction(lock, transaction));
     }
 
     @Override