diff --git a/bj_power_mes/schema/tools/migrate.go b/bj_power_mes/schema/tools/migrate.go
index 25f3b07..9eb5e5f 100644
--- a/bj_power_mes/schema/tools/migrate.go
+++ b/bj_power_mes/schema/tools/migrate.go
@@ -25,16 +25,21 @@ import (
 var (
 	client *ent.Client
 
-	ctx      = context.Background()
+	ctx        = context.Background()
 	configFile = flag.String("config", "etc/bj_power_mes-api.yaml", "配置文件路径")
 )
 
+// migrateConf 与主配置结构一致，只取 Database 段
+type migrateConf struct {
+	Database db.DatabaseConf
+}
+
 func setUp() {
-	var c db.DatabaseConf
+	var c migrateConf
 	conf.MustLoad(*configFile, &c)
 
 	dsn := fmt.Sprintf("host=%s port=%d user=%s password=%s dbname=%s sslmode=disable",
-		c.Host, c.Port, c.User, c.Password, c.Dbname)
+		c.Database.Host, c.Database.Port, c.Database.User, c.Database.Password, c.Database.Dbname)
 	db, err := sql.Open("pgx", dsn)
 	if err != nil {
 		log.Fatal(err)