diff --git a/build.gradle b/build.gradle
index f436299..6ddb752 100644
--- a/build.gradle
+++ b/build.gradle
@@ -5,6 +5,25 @@
     id "com.diffplug.gradle.spotless" version "3.27.0"
 }
 
+sourceSets {
+    main {
+        scala {
+            srcDirs = ['src/main/scala', 'src/main/java']
+        }
+        java {
+            srcDirs = []
+        }
+    }
+    test {
+        scala {
+            srcDirs = ['src/test/scala', 'src/test/java']
+        }
+        java {
+            srcDirs = []
+        }
+    }
+}
+
 repositories {
      mavenLocal()
      mavenCentral()