diff --git a/idea/tests/org/jetbrains/kotlin/idea/hierarchy/AbstractHierarchyWithLibTest.kt b/idea/tests/org/jetbrains/kotlin/idea/hierarchy/AbstractHierarchyWithLibTest.kt
index 1b1471f3372..e27f14db640 100644
--- a/idea/tests/org/jetbrains/kotlin/idea/hierarchy/AbstractHierarchyWithLibTest.kt
+++ b/idea/tests/org/jetbrains/kotlin/idea/hierarchy/AbstractHierarchyWithLibTest.kt
@@ -44,9 +44,10 @@ abstract class AbstractHierarchyWithLibTest : AbstractHierarchyTest() {
                 "// BASE_CLASS: "
         ).singleOrNull() ?: error("File should contain BASE_CLASS directive")
 
-        val targetClass = findTargetJavaClass(directive.trim())
         doHierarchyTest(
                 Computable {
+                    val targetClass = findTargetJavaClass(directive.trim())
+
                     TypeHierarchyTreeStructure(
                             project,
                             targetClass,