diff --git a/tooling/metamodel-generator/hibernate-jpamodelgen.gradle b/tooling/metamodel-generator/hibernate-jpamodelgen.gradle
index 21d1b885af..90bb81ad4d 100644
--- a/tooling/metamodel-generator/hibernate-jpamodelgen.gradle
+++ b/tooling/metamodel-generator/hibernate-jpamodelgen.gradle
@@ -45,6 +45,13 @@ compileTestJava {
     ]
 }
 
+sourceSets {
+    test {
+        java {
+            exclude '**/data/*.java'
+        }
+    }
+}
 
 // Tests with records
 if ( jdkVersions.test.release.asInt() >= 17 && jdkVersions.explicit ) {
@@ -69,7 +76,6 @@ else {
         test {
             java {
                 exclude '**/records/*.java'
-                exclude '**/data/*.java'
             }
         }
     }