diff --git a/plugin-build/primitives-plugin/src/main/kotlin/io/kinference/primitives/PrimitivesGradlePlugin.kt b/plugin-build/primitives-plugin/src/main/kotlin/io/kinference/primitives/PrimitivesGradlePlugin.kt
index d661192..4842e6a 100644
--- a/plugin-build/primitives-plugin/src/main/kotlin/io/kinference/primitives/PrimitivesGradlePlugin.kt
+++ b/plugin-build/primitives-plugin/src/main/kotlin/io/kinference/primitives/PrimitivesGradlePlugin.kt
@@ -23,9 +23,15 @@ class PrimitivesGradlePlugin : Plugin<Project> {
                 primitivesTask.get().dependsOn(task)
             }
 
-            project.tasks.withType(MetadataDependencyTransformationTask::class.java).all {
-                primitivesTask.get().dependsOn(it)
-            }
+//            project.tasks.named { it == "transformDependenciesMetadata" }.all {
+//                primitivesTask.get().dependsOn(it)
+//            }
+//            primitivesTask.get().dependsOn(project.tasks.named("transformDependenciesMetadata"))
+//            project.tasks.withType(MetadataDependencyTransformationTask::class.java).all {
+//                it.transformMetadata()
+//                primitivesTask.get().inputs.files(it.outputs.files).ignoreEmptyDirectories().withPathSensitivity(PathSensitivity.RELATIVE)
+//                primitivesTask.get().dependsOn(it)
+//            }
 
             kotlinExt.sourceSets.all { sourceSet ->
                 val sourceSetName = sourceSet.name
@@ -42,6 +48,8 @@ class PrimitivesGradlePlugin : Plugin<Project> {
             }
 
             fun configureCompilation(compilation: KotlinCompilation<*>) {
+                if (compilation.name == "test") return
+                primitivesTask.get().inputs.files(compilation.compileDependencyFiles)
                 val targetTask = compilation.compileTaskProvider
 
                 targetTask.configure {