diff --git a/benchmark-java/build.sbt b/benchmark-java/build.sbt
index ec3d1f8fd..3613d71cc 100644
--- a/benchmark-java/build.sbt
+++ b/benchmark-java/build.sbt
@@ -29,7 +29,7 @@ val root = project.in(file("."))
       "org.scalatest" %% "scalatest" % "3.1.2" % "test",
       "org.scalatestplus" %% "junit-4-12" % "3.1.2.0" % "test"
     ),
-    Compile / PB.generate := ((Compile / PB.generate) dependsOn (
+    PB.artifactResolver := (PB.artifactResolver dependsOn (
         codeGenProject / Compile / publishLocal)).value
   )
 
diff --git a/project/ReflectiveCodeGen.scala b/project/ReflectiveCodeGen.scala
index 8ce40c1da..27d21e3f4 100644
--- a/project/ReflectiveCodeGen.scala
+++ b/project/ReflectiveCodeGen.scala
@@ -64,8 +64,6 @@ object ReflectiveCodeGen extends AutoPlugin {
             }
           }
         }.value,
-        // Reload generators on each invocation
-        PB.cacheClassLoaders := false,
         setCodeGenerator := loadAndSetGenerator(
           // the magic sauce: use the output classpath from the the sbt-plugin project and instantiate generators from there
           (fullClasspath in Compile in ProjectRef(file("."), "sbt-akka-grpc")).value,
diff --git a/project/plugins.sbt b/project/plugins.sbt
index ceef5c8e4..dea714677 100644
--- a/project/plugins.sbt
+++ b/project/plugins.sbt
@@ -1,6 +1,6 @@
 enablePlugins(BuildInfoPlugin)
 
-val sbtProtocV = "1.0.0"
+val sbtProtocV = "1.0.1"
 
 buildInfoKeys := Seq[BuildInfoKey]("sbtProtocVersion" -> sbtProtocV)