diff --git a/analysis/analysis-api-standalone/src/org/jetbrains/kotlin/analysis/project/structure/builder/KtModuleProviderBuilder.kt b/analysis/analysis-api-standalone/src/org/jetbrains/kotlin/analysis/project/structure/builder/KtModuleProviderBuilder.kt
index 98e7e6a5b9e..03f81e94d9c 100644
--- a/analysis/analysis-api-standalone/src/org/jetbrains/kotlin/analysis/project/structure/builder/KtModuleProviderBuilder.kt
+++ b/analysis/analysis-api-standalone/src/org/jetbrains/kotlin/analysis/project/structure/builder/KtModuleProviderBuilder.kt
@@ -15,12 +15,13 @@ import kotlin.contracts.InvocationKind
 import kotlin.contracts.contract
 
 public class KtModuleProviderBuilder(
-    public val kotlinCoreProjectEnvironment: KotlinCoreProjectEnvironment
+    public val kotlinCoreProjectEnvironment: KotlinCoreProjectEnvironment,
 ) {
     private val mainModules: MutableList<KtModule> = mutableListOf()
 
-    public fun addModule(module: KtModule) {
+    public fun <M : KtModule> addModule(module: M): M {
         mainModules.add(module)
+        return module
     }
 
     public lateinit var platform: TargetPlatform