diff --git a/.github/workflows/test.yaml b/.github/workflows/test.yaml
index fcf71df..065a7de 100644
--- a/.github/workflows/test.yaml
+++ b/.github/workflows/test.yaml
@@ -8,7 +8,6 @@ on:
     branches: ["*"]
 env:
   GRPC_VERSION: "1.62"
-  BUF_VERSION: "1.30.0"
 jobs:
   pytest:
     name: "Unit and Integration Tests"
diff --git a/buf.gen.yaml b/buf.gen.yaml
index c1126a5..f50d9da 100755
--- a/buf.gen.yaml
+++ b/buf.gen.yaml
@@ -1,13 +1,14 @@
-#!/usr/bin/env -S buf generate buf.build/authzed/api:v1.35.0 --template
+#!/usr/bin/env -S buf generate
 ---
-version: "v1"
+version: "v2"
+inputs:
+  - module: "buf.build/authzed/api:v1.35.0"
 plugins:
-  - name: "python"
+  - local: "protoc-gen-grpc"
     out: "."
-  - name: "grpc-python"
+  - local: "protoc-gen-grpc_python_plugin"
     out: "."
-    path: "grpc_python_plugin"
-  - name: "mypy"
+  - local: "protoc-gen-mypy"
     out: "."
-  - name: "mypy_grpc"
+  - local: "protoc-gen-mypy_grpc"
     out: "."