diff --git a/buildSrc/src/main/kotlin/Deployment.kt b/buildSrc/src/main/kotlin/Deployment.kt
index 11698ee4..601ceace 100644
--- a/buildSrc/src/main/kotlin/Deployment.kt
+++ b/buildSrc/src/main/kotlin/Deployment.kt
@@ -1,6 +1,7 @@
 @file:Suppress("DEPRECATION")
 
 import com.android.build.gradle.LibraryExtension
+import com.android.build.gradle.internal.api.DefaultAndroidSourceDirectorySet
 import org.gradle.api.Action
 import org.gradle.api.Project
 import org.gradle.api.plugins.JavaPluginConvention
@@ -64,7 +65,7 @@ object Deployment {
                 val main = android.sourceSets.getByName("main")
                 val sourcesJar by project.tasks.creating(Jar::class) {
                     archiveClassifier.set("sources")
-                    from(main.java.srcDirs)
+                    from((main.kotlin as DefaultAndroidSourceDirectorySet).srcDirs)
                 }
                 val javadocJar by project.tasks.creating(Jar::class) {
                     archiveClassifier.set("javadoc")
diff --git a/compose/build.gradle.kts b/compose/build.gradle.kts
index aae49593..2e2a5ef5 100644
--- a/compose/build.gradle.kts
+++ b/compose/build.gradle.kts
@@ -16,11 +16,11 @@ android {
 
     sourceSets {
         getByName("main") {
-            java.srcDir("src/main/kotlin")
+            kotlin.srcDirs("src/main/kotlin")
             res.srcDir("src/main/res")
         }
         getByName("test") {
-            java.srcDir("src/test/kotlin")
+            kotlin.srcDirs("src/test/kotlin")
         }
     }
 
diff --git a/sample/build.gradle.kts b/sample/build.gradle.kts
index 94c63ac0..3d15f783 100644
--- a/sample/build.gradle.kts
+++ b/sample/build.gradle.kts
@@ -53,11 +53,11 @@ android {
 
     sourceSets {
         getByName("main") {
-            java.srcDir("src/main/kotlin")
+            kotlin.srcDirs("src/main/kotlin")
             res.srcDir("src/main/res")
         }
         getByName("androidTest") {
-            java.srcDir("src/androidTest/kotlin")
+            kotlin.srcDirs("src/androidTest/kotlin")
         }
     }
 }