diff --git a/prepare/analysis-api-test-framework/build.gradle.kts b/prepare/analysis-api-test-framework/build.gradle.kts
index d9eb642a1df..fd16caa569f 100644
--- a/prepare/analysis-api-test-framework/build.gradle.kts
+++ b/prepare/analysis-api-test-framework/build.gradle.kts
@@ -8,20 +8,29 @@ val testModules = listOf(
     ":analysis:analysis-test-framework",
     ":analysis:analysis-api-impl-barebone",
     ":analysis:analysis-api-impl-base",
-    ":analysis:analysis-api-standalone"
+    ":analysis:analysis-api-standalone",
+    ":analysis:decompiled:decompiler-to-file-stubs",
+)
+
+val mainModules = listOf(
+    ":kotlin-preloader",
 )
 
 dependencies {
     testModules.forEach {
         embedded(projectTests(it)) { isTransitive = false }
     }
+
+    mainModules.forEach {
+        embedded(project(it)) { isTransitive = false }
+    }
 }
 
 publish()
 runtimeJar()
 sourcesJar {
     from {
-        testModules.map { project(it).testSourceSet.allSource }
+        mainModules.map { project(it).mainSourceSet.allSource } + testModules.map { project(it).testSourceSet.allSource }
     }
 }
 javadocJar()