diff --git a/database/new.go b/database/new.go
index 4baf740..815fbd6 100644
--- a/database/new.go
+++ b/database/new.go
@@ -15,8 +15,6 @@ import (
 var (
 	// MigrateTables holds the tables that need to be auto-migrated on database initialization
 	MigrateTables []any
-	// Init is an optional initialization function that can be executed after database connection is established
-	InitFunc *func() error = nil
 )
 
 // NewDatabase creates a new database connection based on the provided driver type
diff --git a/service/service.go b/service/service.go
index b7aafff..19b1aa6 100644
--- a/service/service.go
+++ b/service/service.go
@@ -108,6 +108,17 @@ func (s *Service) Gateway(grpcAddr string, httpAddr string) {
 	http.ListenAndServe(httpAddr, s.Opts.GatewayMux)
 }
 
+func (s *Service) Use(initFunc *func() error) {
+	// Execute the Init function if it's not nil
+	if initFunc != nil {
+		err := (*initFunc)()
+		if err != nil {
+			print.Error(err.Error())
+			panic(err)
+		}
+	}
+}
+
 func (s *Service) Stop() {
 	s.GrpcSrv.GracefulStop()
 }
diff --git a/with/init.go b/with/init.go
deleted file mode 100644
index 1b60b68..0000000
--- a/with/init.go
+++ /dev/null
@@ -1,18 +0,0 @@
-package with
-
-import (
-	"git.apinb.com/bsm-sdk/core/database"
-	"git.apinb.com/bsm-sdk/core/errcode"
-	"git.apinb.com/bsm-sdk/core/print"
-)
-
-func InitData() {
-	// Execute the Init function if it's not nil
-	if database.InitFunc != nil {
-		err := (*database.InitFunc)()
-		if err != nil {
-			print.Error(errcode.ErrEtcd.Error())
-			panic(err)
-		}
-	}
-}