diff --git a/build.gradle b/build.gradle
index 830522c64f..15d326d717 100644
--- a/build.gradle
+++ b/build.gradle
@@ -210,7 +210,11 @@ configure(javaProjects) { subproject ->
                 srcDirs = ['src/test/resources', 'src/test/java']
             }
         }
-        feature
+        feature {
+            java {
+                srcDir 'src/main/java'
+            }
+        }
     }
 
     java {