diff --git a/compiler/ir/ir.tests/tests/org/jetbrains/kotlin/ir/AbstractIrGeneratorTestCase.kt b/compiler/ir/ir.tests/tests/org/jetbrains/kotlin/ir/AbstractIrGeneratorTestCase.kt
index f49ebc3e1b6..53c6726f3bf 100644
--- a/compiler/ir/ir.tests/tests/org/jetbrains/kotlin/ir/AbstractIrGeneratorTestCase.kt
+++ b/compiler/ir/ir.tests/tests/org/jetbrains/kotlin/ir/AbstractIrGeneratorTestCase.kt
@@ -41,7 +41,8 @@ abstract class AbstractIrGeneratorTestCase : CodegenTestCase() {
         AnalyzingUtils.throwExceptionOnErrors(analysisResult.bindingContext)
         val psi2ir = Psi2IrTranslator()
         val irModule = psi2ir.generateModule(analysisResult.moduleDescriptor, myFiles.psiFiles, analysisResult.bindingContext)
-        return testFiles.filter { it.name.endsWith(".kt") }.zip(irModule.files).toMap()
+        val ktFiles = testFiles.filter { it.name.endsWith(".kt") }
+        return ktFiles.zip(irModule.files).toMap()
     }
 }