1 files changed, 4 insertions, 6 deletions
diff --git a/config-model/src/test/java/com/yahoo/vespa/model/application/validation/change/RestartChangesDefersConfigChangesTest.java b/config-model/src/test/java/com/yahoo/vespa/model/application/validation/change/RestartChangesDefersConfigChangesTest.java
index fda4c54c154..35aa9a3c988 100644
--- a/config-model/src/test/java/com/yahoo/vespa/model/application/validation/change/RestartChangesDefersConfigChangesTest.java
+++ b/config-model/src/test/java/com/yahoo/vespa/model/application/validation/change/RestartChangesDefersConfigChangesTest.java
@@ -1,11 +1,9 @@
 // Copyright 2017 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
 package com.yahoo.vespa.model.application.validation.change;
 
-import com.yahoo.cloud.config.log.LogdConfig;
 import com.yahoo.config.model.provision.InMemoryProvisioner;
 import com.yahoo.config.provision.Environment;
 import com.yahoo.config.provision.NodeResources;
-import com.yahoo.container.ComponentsConfig;
 import com.yahoo.container.QrConfig;
 import com.yahoo.vespa.model.VespaModel;
 import com.yahoo.vespa.model.application.validation.ValidationTester;
@@ -28,15 +26,15 @@ public class RestartChangesDefersConfigChangesTest {
 
         // Change node count - no restart
         VespaModel gen2 = tester.deploy(gen1, getServices(4, 3), Environment.prod, null).getFirst();
-        var config2 = new ComponentsConfig.Builder();
+        var config2 = new QrConfig.Builder();
         gen2.getContainerClusters().get("default").getContainers().get(0).getConfig(config2);
-        assertFalse(config2.getApplyOnRestart());
+        assertFalse(config2.build().restartOnDeploy());
 
         // Change memory amount - requires restart
         VespaModel gen3 = tester.deploy(gen2, getServices(4, 2), Environment.prod, null).getFirst();
-        var config3 = new ComponentsConfig.Builder();
+        var config3 = new QrConfig.Builder();
         gen3.getContainerClusters().get("default").getContainers().get(0).getConfig(config3);
-        assertTrue(config3.getApplyOnRestart());
+        assertTrue(config3.build().restartOnDeploy());
     }
 
     private static String getServices(int nodes, int memory) {