diff --git a/libraries/tools/kotlin-gradle-plugin/src/common/kotlin/org/jetbrains/kotlin/gradle/plugin/mpp/MetadataDependencyTransformationTask.kt b/libraries/tools/kotlin-gradle-plugin/src/common/kotlin/org/jetbrains/kotlin/gradle/plugin/mpp/MetadataDependencyTransformationTask.kt
index 3e0edebdb8e..97b765beb5e 100644
--- a/libraries/tools/kotlin-gradle-plugin/src/common/kotlin/org/jetbrains/kotlin/gradle/plugin/mpp/MetadataDependencyTransformationTask.kt
+++ b/libraries/tools/kotlin-gradle-plugin/src/common/kotlin/org/jetbrains/kotlin/gradle/plugin/mpp/MetadataDependencyTransformationTask.kt
@@ -48,15 +48,12 @@ open class MetadataDependencyTransformationTask
         project.kotlinTransformedMetadataLibraryDirectoryForBuild(kotlinSourceSet.name)
     }
 
-    // TODO: Repalce with sourceSet metadata
     @Suppress("unused") // Gradle input
-    @get:IgnoreEmptyDirectories
-    @get:NormalizeLineEndings
     @get:InputFiles
     @get:PathSensitive(PathSensitivity.RELATIVE)
-    internal val allSourceSetsMetadataConfiguration: FileCollection by lazy {
-        project.files(project.configurations.getByName(ALL_COMPILE_METADATA_CONFIGURATION_NAME))
-    }
+    @get:IgnoreEmptyDirectories
+    @get:NormalizeLineEndings
+    internal val configurationToResolve: FileCollection get() = kotlinSourceSet.internal.resolvableMetadataConfiguration
 
     private val participatingSourceSets: Set<KotlinSourceSet>
         get() = transformation.kotlinSourceSet.internal.withDependsOnClosure.toMutableSet().apply {