diff --git a/build.gradle.kts b/build.gradle.kts
index 1cf6ebd3ff0..a9d55b293eb 100644
--- a/build.gradle.kts
+++ b/build.gradle.kts
@@ -759,9 +759,10 @@ fun Project.configureShadowJarSubstitutionInCompileClasspath() {
     }
 
     sourceSets.all {
-        val compileClasspathConfig = configurations.getByName(compileClasspathConfigurationName)
-        compileClasspathConfig.resolutionStrategy.dependencySubstitution {
-            all(::configureSubstitution)
+        for (configName in listOf(compileOnlyConfigurationName, compileClasspathConfigurationName)) {
+            configurations.getByName(configName).resolutionStrategy.dependencySubstitution {
+                all(::configureSubstitution)
+            }
         }
     }
 }