diff --git a/build.gradle b/build.gradle
index 8fef212719..830522c64f 100644
--- a/build.gradle
+++ b/build.gradle
@@ -210,16 +210,17 @@ configure(javaProjects) { subproject ->
                 srcDirs = ['src/test/resources', 'src/test/java']
             }
         }
+        feature
     }
 
     java {
         withJavadocJar()
         withSourcesJar()
         registerFeature('optional') {
-            usingSourceSet(sourceSets.main)
+            usingSourceSet(sourceSets.feature)
         }
         registerFeature('provided') {
-            usingSourceSet(sourceSets.main)
+            usingSourceSet(sourceSets.feature)
         }
     }