diff --git a/buildSrc/src/main/kotlin/KiteUiPlugin.kt b/buildSrc/src/main/kotlin/KiteUiPlugin.kt
index 7025a918..8c45f30e 100644
--- a/buildSrc/src/main/kotlin/KiteUiPlugin.kt
+++ b/buildSrc/src/main/kotlin/KiteUiPlugin.kt
@@ -172,7 +172,7 @@ class KiteUiPlugin : Plugin<Project> {
                 generateAutoroutes(sources, out)
             }
             afterEvaluate {
-                tasks.findByName("compileKotlinMetadata")?.dependsOn(task)
+                tasks.filter { it.name.contains("compileKotlin") }.forEach { it.dependsOn(task) }
             }
         }
 
diff --git a/gradle-plugin/src/main/kotlin/KiteUiPlugin.kt b/gradle-plugin/src/main/kotlin/KiteUiPlugin.kt
index 7025a918..8c45f30e 100644
--- a/gradle-plugin/src/main/kotlin/KiteUiPlugin.kt
+++ b/gradle-plugin/src/main/kotlin/KiteUiPlugin.kt
@@ -172,7 +172,7 @@ class KiteUiPlugin : Plugin<Project> {
                 generateAutoroutes(sources, out)
             }
             afterEvaluate {
-                tasks.findByName("compileKotlinMetadata")?.dependsOn(task)
+                tasks.filter { it.name.contains("compileKotlin") }.forEach { it.dependsOn(task) }
             }
         }