1 files changed, 12 insertions, 12 deletions
diff --git a/config-model/src/main/java/com/yahoo/vespa/model/container/ml/ModelsEvaluatorTester.java b/config-model/src/main/java/com/yahoo/vespa/model/container/ml/ModelsEvaluatorTester.java
index 800e766706b..bc936dc7441 100644
--- a/config-model/src/main/java/com/yahoo/vespa/model/container/ml/ModelsEvaluatorTester.java
+++ b/config-model/src/main/java/com/yahoo/vespa/model/container/ml/ModelsEvaluatorTester.java
@@ -115,27 +115,27 @@ public class ModelsEvaluatorTester {
     }
 
     private static RankProfilesConfig getRankProfilesConfig(RankProfileList rankProfileList) {
-        RankProfilesConfig.Builder builder = new RankProfilesConfig.Builder();
-        rankProfileList.getConfig(builder);
-        return builder.build();
+        return new RankProfilesConfig.Builder()
+                .rankprofile(rankProfileList.getRankProfilesConfig())
+                .build();
     }
 
     private static RankingConstantsConfig getRankingConstantConfig(RankProfileList rankProfileList) {
-        RankingConstantsConfig.Builder builder = new RankingConstantsConfig.Builder();
-        rankProfileList.getConfig(builder);
-        return builder.build();
+        return new RankingConstantsConfig.Builder()
+                .constant(rankProfileList.getConstantsConfig())
+                .build();
     }
 
     private static RankingExpressionsConfig getRankingExpressionsConfig(RankProfileList rankProfileList) {
-        RankingExpressionsConfig.Builder builder = new RankingExpressionsConfig.Builder();
-        rankProfileList.getConfig(builder);
-        return builder.build();
+        return new RankingExpressionsConfig.Builder()
+                .expression(rankProfileList.getExpressionsConfig())
+                .build();
     }
 
     private static OnnxModelsConfig getOnnxModelsConfig(RankProfileList rankProfileList) {
-        OnnxModelsConfig.Builder builder = new OnnxModelsConfig.Builder();
-        rankProfileList.getConfig(builder);
-        return builder.build();
+        return new OnnxModelsConfig.Builder()
+                .model(rankProfileList.getOnnxConfig())
+                .build();
     }
 
     private static FileAcquirer createFileAcquirer(MockFileRegistry fileRegistry, File appDir) {