diff --git a/HIRS_AttestationCA/build.gradle b/HIRS_AttestationCA/build.gradle
index 31806ddd..04a80074 100644
--- a/HIRS_AttestationCA/build.gradle
+++ b/HIRS_AttestationCA/build.gradle
@@ -1,6 +1,7 @@
 plugins {
     id 'java'
     id 'io.spring.dependency-management' version '1.1.0'
+    id 'com.google.protobuf' version '0.9.4'
 }
 
 java {
@@ -50,13 +51,18 @@ dependencies {
     annotationProcessor libs.lombok
 }
 
-task generateProtoBuf(type:Exec) {
-    workingDir 'config'
-
-    commandLine './genJavaProtoBuf.sh'
+protobuf {
+  protoc {
+    artifact = 'com.google.protobuf:protoc:3.24.3'
+  }
+}
+sourceSets {
+  main {
+    proto {
+      srcDir '../HIRS_ProvisionerTPM2/src'
+    }
+  }
 }
-
-compileJava.dependsOn generateProtoBuf
 
 test {
     useJUnitPlatform()