diff --git a/build.gradle b/build.gradle
index e47374a..27fc1a7 100644
--- a/build.gradle
+++ b/build.gradle
@@ -6,6 +6,23 @@ plugins {
     id "signing"
 }
 
+sourceSets {
+    main {
+        java {
+            srcDirs 'src'
+        }
+    }
+    test {
+        java {
+            srcDirs 'test/functional'
+            srcDirs 'test/integration'
+            srcDirs 'test/unit'
+        }
+    }
+
+
+}
+
 task sourceJar(type: Jar) {
     archiveClassifier = "sources"
     from sourceSets.main.allJava