1 files changed, 5 insertions, 5 deletions

diff --git a/vespa-documentgen-plugin/src/main/java/com/yahoo/vespa/DocumentGenMojo.java b/vespa-documentgen-plugin/src/main/java/com/yahoo/vespa/DocumentGenMojo.java
index 3a4b36d9bb5..79325075a42 100644
--- a/vespa-documentgen-plugin/src/main/java/com/yahoo/vespa/DocumentGenMojo.java
+++ b/vespa-documentgen-plugin/src/main/java/com/yahoo/vespa/DocumentGenMojo.java
@@ -18,7 +18,7 @@ import com.yahoo.document.annotation.AnnotationType;
 import com.yahoo.documentmodel.NewDocumentType;
 import com.yahoo.documentmodel.VespaDocumentType;
 import com.yahoo.searchdefinition.Schema;
-import com.yahoo.searchdefinition.SearchBuilder;
+import com.yahoo.searchdefinition.SchemaBuilder;
 import com.yahoo.searchdefinition.document.FieldSet;
 import com.yahoo.searchdefinition.parser.ParseException;
 import org.apache.maven.plugin.AbstractMojo;
@@ -114,7 +114,7 @@ public class DocumentGenMojo extends AbstractMojo {
         annotationTypes = new HashMap<>();
 
         outputDir.mkdirs();
-        SearchBuilder builder = buildSearches(schemasDir);
+        SchemaBuilder builder = buildSearches(schemasDir);
 
         boolean annotationsExported=false;
         for (NewDocumentType docType : builder.getModel().getDocumentManager().getTypes()) {
@@ -134,9 +134,9 @@ public class DocumentGenMojo extends AbstractMojo {
         if (project!=null) project.addCompileSourceRoot(outputDirectory.toString());
     }
 
-    private SearchBuilder buildSearches(File sdDir) {
+    private SchemaBuilder buildSearches(File sdDir) {
         File[] sdFiles = sdDir.listFiles((dir, name) -> name.endsWith(".sd"));
-        SearchBuilder builder = new SearchBuilder(true);
+        SchemaBuilder builder = new SchemaBuilder(true);
         for (File f : sdFiles) {
             try {
                 long modTime = f.lastModified();
@@ -149,7 +149,7 @@ public class DocumentGenMojo extends AbstractMojo {
             }
         }
         builder.build();
-        for (Schema schema : builder.getSearchList() ) {
+        for (Schema schema : builder.getSchemaList() ) {
             this.searches.put(schema.getName(), schema);
         }
         return builder;