diff --git a/go.mod b/go.mod
index 8247eb3..fac14d1 100644
--- a/go.mod
+++ b/go.mod
@@ -6,7 +6,7 @@ toolchain go1.25.6
 
 require (
 	github.com/Masterminds/squirrel v1.5.4
-	github.com/goravel/framework v1.16.1-0.20260128144145-280cdc17fcee
+	github.com/goravel/framework v1.16.1-0.20260202061217-60e5444dde33
 	github.com/spf13/cast v1.10.0
 	github.com/stretchr/testify v1.11.1
 	gorm.io/driver/postgres v1.6.0
diff --git a/go.sum b/go.sum
index ccff64c..6b86e2a 100644
--- a/go.sum
+++ b/go.sum
@@ -80,8 +80,8 @@ github.com/gookit/color v1.4.2/go.mod h1:fqRyamkC1W8uxl+lxCQxOT09l/vYfZ+QeiX3rKQ
 github.com/gookit/color v1.5.0/go.mod h1:43aQb+Zerm/BWh2GnrgOQm7ffz7tvQXEKV6BFMl7wAo=
 github.com/gookit/color v1.5.4 h1:FZmqs7XOyGgCAxmWyPslpiok1k05wmY3SJTytgvYFs0=
 github.com/gookit/color v1.5.4/go.mod h1:pZJOeOS8DM43rXbp4AZo1n9zCU2qjpcRko0b6/QJi9w=
-github.com/goravel/framework v1.16.1-0.20260128144145-280cdc17fcee h1:J/tYlCECxU+x2xwG8xuwVzRUoZsIRj+3LGldwAjVRyw=
-github.com/goravel/framework v1.16.1-0.20260128144145-280cdc17fcee/go.mod h1:RvgW+VhSE0aCr2ju/d6B8PsPsz04Q7gUFnWCY9ru6MI=
+github.com/goravel/framework v1.16.1-0.20260202061217-60e5444dde33 h1:i95pSIYcnbCn/ZSEKTHOrtqxzCdbnQAQV1QWesRVWV0=
+github.com/goravel/framework v1.16.1-0.20260202061217-60e5444dde33/go.mod h1:RvgW+VhSE0aCr2ju/d6B8PsPsz04Q7gUFnWCY9ru6MI=
 github.com/jackc/pgpassfile v1.0.0 h1:/6Hmqy13Ss2zCq62VdNG8tM1wchn8zjSGOBJ6icpsIM=
 github.com/jackc/pgpassfile v1.0.0/go.mod h1:CEx0iS5ambNFdcRtxPj5JhEz+xB6uRky5eyVu/W2HEg=
 github.com/jackc/pgservicefile v0.0.0-20240606120523-5a60cdf6a761 h1:iCEnooe7UlwOQYpKFhBabPMi4aNAfoODPEFNiAnClxo=
diff --git a/setup/setup.go b/setup/setup.go
index c312a42..218196d 100755
--- a/setup/setup.go
+++ b/setup/setup.go
@@ -48,7 +48,7 @@ func main() {
 		// Add postgres service provider to providers.go if using bootstrap setup
 		modify.When(func(_ map[string]any) bool {
 			return env.IsBootstrapSetup()
-		}, modify.AddProviderApply(moduleImport, postgresServiceProvider)),
+		}, modify.RegisterProvider(moduleImport, postgresServiceProvider)),
 
 		// Add postgres connection to database.go
 		modify.GoFile(databaseConfigPath).Find(match.Imports()).Modify(
@@ -77,6 +77,6 @@ func main() {
 		// Remove postgres service provider from providers.go if using bootstrap setup
 		modify.When(func(_ map[string]any) bool {
 			return env.IsBootstrapSetup()
-		}, modify.RemoveProviderApply(moduleImport, postgresServiceProvider)),
+		}, modify.UnregisterProvider(moduleImport, postgresServiceProvider)),
 	).Execute()
 }