diff --git a/idea/idea-jvm/src/org/jetbrains/kotlin/idea/run/KotlinJUnitRunConfigurationProducer.kt b/idea/idea-jvm/src/org/jetbrains/kotlin/idea/run/KotlinJUnitRunConfigurationProducer.kt
index d6a0eff4578..3ee88706f4f 100644
--- a/idea/idea-jvm/src/org/jetbrains/kotlin/idea/run/KotlinJUnitRunConfigurationProducer.kt
+++ b/idea/idea-jvm/src/org/jetbrains/kotlin/idea/run/KotlinJUnitRunConfigurationProducer.kt
@@ -120,6 +120,7 @@ class KotlinJUnitRunConfigurationProducer : RunConfigurationProducer<JUnitConfig
         val implModule = ktFile.module?.findJvmImplementationModule() ?: return
         val sdk = ModuleRootManager.getInstance(implModule).sdk
         if (sdk != null) {
+            configuration.setModule(implModule)
             configuration.isAlternativeJrePathEnabled = true
             configuration.alternativeJrePath = sdk.homePath
         }