diff --git a/wx-official-orm/src/test/java/cn/skcks/wx/official/orm/OrmGenerator.java b/wx-official-orm/src/test/java/cn/skcks/wx/official/orm/OrmGenerator.java
index 5a7b5fb..8ab8998 100644
--- a/wx-official-orm/src/test/java/cn/skcks/wx/official/orm/OrmGenerator.java
+++ b/wx-official-orm/src/test/java/cn/skcks/wx/official/orm/OrmGenerator.java
@@ -8,6 +8,7 @@ import lombok.SneakyThrows;
 import org.springframework.boot.SpringApplication;
 import org.springframework.boot.WebApplicationType;
 import org.springframework.boot.autoconfigure.SpringBootApplication;
+import org.springframework.boot.builder.SpringApplicationBuilder;
 
 import javax.sql.DataSource;
 import java.io.BufferedReader;
@@ -18,9 +19,10 @@ import java.util.Map;
 @SpringBootApplication
 public class OrmGenerator {
     public static void main(String[] args) {
-        SpringApplication springApplication = new SpringApplication(OrmGenerator.class);
-        springApplication.setAdditionalProfiles("generator");
-        springApplication.setWebApplicationType(WebApplicationType.NONE);
+        SpringApplication springApplication = new SpringApplicationBuilder(OrmGenerator.class)
+                .profiles("generator")
+                .web(WebApplicationType.NONE)
+                .build();
         springApplication.run(args);
         generate();
     }
diff --git a/wx-official-orm/src/test/java/cn/skcks/wx/official/schema/SchemaGenerator.java b/wx-official-orm/src/test/java/cn/skcks/wx/official/schema/SchemaGenerator.java
index a46efb8..dfeaf70 100644
--- a/wx-official-orm/src/test/java/cn/skcks/wx/official/schema/SchemaGenerator.java
+++ b/wx-official-orm/src/test/java/cn/skcks/wx/official/schema/SchemaGenerator.java
@@ -3,13 +3,15 @@ package cn.skcks.wx.official.schema;
 import org.springframework.boot.SpringApplication;
 import org.springframework.boot.WebApplicationType;
 import org.springframework.boot.autoconfigure.SpringBootApplication;
+import org.springframework.boot.builder.SpringApplicationBuilder;
 
 @SpringBootApplication
 public class SchemaGenerator {
     public static void main(String[] args) {
-        SpringApplication springApplication = new SpringApplication(SchemaGenerator.class);
-        springApplication.setAdditionalProfiles("schema");
-        springApplication.setWebApplicationType(WebApplicationType.NONE);
+        SpringApplication springApplication = new SpringApplicationBuilder(SchemaGenerator.class)
+                .profiles("schema")
+                .web(WebApplicationType.NONE)
+                .build();
         springApplication.run(args);
     }
 }