diff --git a/java/client/build.gradle b/java/client/build.gradle
index 28bd8fcdc..21002dd5b 100644
--- a/java/client/build.gradle
+++ b/java/client/build.gradle
@@ -26,17 +26,17 @@ sourceSets {
             srcDir '../../proto'
         }
     }
+    test {
+        java {
+            srcDir '../shared/test/java'
+        }
+    }
 }
 
 protobuf {
     protoc {
         artifact = 'com.google.protobuf:protoc:3.24.4'
     }
-    test {
-        java {
-            srcDir '../shared/test/java'
-        }
-    }
 }
 
 dependencies {
diff --git a/java/server/build.gradle b/java/server/build.gradle
index 4f3196e5b..4b04bc947 100644
--- a/java/server/build.gradle
+++ b/java/server/build.gradle
@@ -26,17 +26,17 @@ sourceSets {
             srcDir '../../proto'
         }
     }
+    test {
+        java {
+            srcDir '../shared/test/java'
+        }
+    }
 }
 
 protobuf {
     protoc {
         artifact = 'com.google.protobuf:protoc:3.24.4'
     }
-    test {
-        java {
-            srcDir '../shared/test/java'
-        }
-    }
 }
 
 dependencies {