diff --git a/internal/setup/sql_upgrade_test.go b/internal/setup/sql_upgrade_test.go
index c4379d87..6a5f1994 100644
--- a/internal/setup/sql_upgrade_test.go
+++ b/internal/setup/sql_upgrade_test.go
@@ -6,7 +6,11 @@ import (
 )
 
 func TestUpgradeSQLData(t *testing.T) {
-	db, err := dbs.Default()
+	db, err := dbs.NewInstanceFromConfig(&dbs.DBConfig{
+		Driver: "mysql",
+		Dsn:    "root:123456@tcp(127.0.0.1:3306)/db_edge_new?charset=utf8mb4&timeout=30s",
+		Prefix: "edge",
+	})
 	if err != nil {
 		t.Fatal(err)
 	}