diff --git a/.github/workflows/release.yml b/.github/workflows/release.yml
index 5b11ea2..3248687 100644
--- a/.github/workflows/release.yml
+++ b/.github/workflows/release.yml
@@ -43,6 +43,8 @@ jobs:
         working-directory: ./provider
         run: go mod download
 
+      - run: make tfgen
+
       - run: make build_nodejs
 
       - uses: goreleaser/goreleaser-action@v5
diff --git a/provider/cmd/pulumi-resource-planetscale/main.go b/provider/cmd/pulumi-resource-planetscale/main.go
index 632e2c1..fda605a 100644
--- a/provider/cmd/pulumi-resource-planetscale/main.go
+++ b/provider/cmd/pulumi-resource-planetscale/main.go
@@ -15,11 +15,11 @@
 package main
 
 import (
+	"context"
 	_ "embed"
 
-	"github.com/pulumi/pulumi-terraform-bridge/v3/pkg/tfbridge"
+	"github.com/pulumi/pulumi-terraform-bridge/pf/tfbridge"
 	planetscale "github.com/sst/pulumi-planetscale/provider"
-	"github.com/sst/pulumi-planetscale/provider/pkg/version"
 )
 
 //go:embed schema.json
@@ -27,5 +27,6 @@ var pulumiSchema []byte
 
 func main() {
 	// Modify the path to point to the new provider
-	tfbridge.Main("planetscale", version.Version, planetscale.Provider(), pulumiSchema)
+	tfbridge.Main(context.Background(), "planetscale", planetscale.Provider(),
+		tfbridge.ProviderMetadata{PackageSchema: pulumiSchema})
 }
diff --git a/provider/cmd/pulumi-tfgen-planetscale/main.go b/provider/cmd/pulumi-tfgen-planetscale/main.go
index 2b85756..27635ee 100644
--- a/provider/cmd/pulumi-tfgen-planetscale/main.go
+++ b/provider/cmd/pulumi-tfgen-planetscale/main.go
@@ -15,12 +15,11 @@
 package main
 
 import (
-	"github.com/pulumi/pulumi-terraform-bridge/v3/pkg/tfgen"
+	"github.com/pulumi/pulumi-terraform-bridge/pf/tfgen"
 	planetscale "github.com/sst/pulumi-planetscale/provider"
-	"github.com/sst/pulumi-planetscale/provider/pkg/version"
 )
 
 func main() {
 	// Modify the path to point to the new provider
-	tfgen.Main("planetscale", version.Version, planetscale.Provider())
+	tfgen.Main("planetscale", planetscale.Provider())
 }