diff --git a/build-logic/commons/src/main/groovy/bisq.protobuf.gradle b/build-logic/commons/src/main/groovy/bisq.protobuf.gradle
index 51b1816d5a..cd8e0cbda6 100644
--- a/build-logic/commons/src/main/groovy/bisq.protobuf.gradle
+++ b/build-logic/commons/src/main/groovy/bisq.protobuf.gradle
@@ -8,12 +8,18 @@ dependencies {
     implementation libs.protobuf.java
 }
 
-ext {
-    generatedProtoSrcDir = 'build/generated/source/main/java'
-}
+sourceSets {
+    main {
+        java {
+            srcDirs += ['build/generated/source/main/java']
+        }
+    }
 
-sourceSets.main {
-    java.srcDirs += [generatedProtoSrcDir]
+    test {
+        java {
+            srcDirs += ['build/generated/source/test/java']
+        }
+    }
 }
 
 ext {