diff --git a/idea/src/org/jetbrains/kotlin/idea/run/KotlinJUnitRunConfigurationProducer.kt b/idea/src/org/jetbrains/kotlin/idea/run/KotlinJUnitRunConfigurationProducer.kt
index 015201848b5..e6e387ca2ea 100644
--- a/idea/src/org/jetbrains/kotlin/idea/run/KotlinJUnitRunConfigurationProducer.kt
+++ b/idea/src/org/jetbrains/kotlin/idea/run/KotlinJUnitRunConfigurationProducer.kt
@@ -130,9 +130,10 @@ public class KotlinJUnitRunConfigurationProducer : RunConfigurationProducer<JUni
     }
 
     private fun getTestClass(leaf: PsiElement): PsiClass? {
+        val containingFile = leaf.getContainingFile() as? JetFile ?: return null
         var jetClass = leaf.getParentOfType<JetClass>(false)
         if (!jetClass.isJUnitTestClass()) {
-            jetClass = getTestClassInFile(leaf.getContainingFile() as JetFile)
+            jetClass = getTestClassInFile(containingFile)
         }
         if (jetClass != null) {
             return LightClassUtil.getPsiClass(jetClass)