diff --git a/idea/src/org/jetbrains/kotlin/idea/run/KotlinRunConfigurationProducer.kt b/idea/src/org/jetbrains/kotlin/idea/run/KotlinRunConfigurationProducer.kt
index b9c3d1aa168..beee59b38b3 100644
--- a/idea/src/org/jetbrains/kotlin/idea/run/KotlinRunConfigurationProducer.kt
+++ b/idea/src/org/jetbrains/kotlin/idea/run/KotlinRunConfigurationProducer.kt
@@ -45,7 +45,7 @@ public class KotlinRunConfigurationProducer : RunConfigurationProducer<JetRunCon
                                                context: ConfigurationContext,
                                                sourceElement: Ref<PsiElement>): Boolean {
         val location = context.getLocation() ?: return false
-        val module = context.getModule() ?: return false
+        val module = location.getModule() ?: return false
         val container = getEntryPointContainer(location)
         val startClassFQName = getStartClassFqName(container) ?: return false