diff --git a/buildSrc/plugins/src/main/kotlin/org/jetbrains/kotlin/FrameworkTest.kt b/buildSrc/plugins/src/main/kotlin/org/jetbrains/kotlin/FrameworkTest.kt
index 6a89037f6cc..6be99b07102 100644
--- a/buildSrc/plugins/src/main/kotlin/org/jetbrains/kotlin/FrameworkTest.kt
+++ b/buildSrc/plugins/src/main/kotlin/org/jetbrains/kotlin/FrameworkTest.kt
@@ -32,9 +32,7 @@ open class FrameworkTest : DefaultTask() {
     lateinit var frameworkName: String
 
     private val testOutput by lazy {
-        ((project.findProperty("sourceSets") as SourceSetContainer)
-                .getByName("testOutputFramework") as SourceSet).output.dirs.singleFile.absolutePath
-                ?: throw RuntimeException("Empty sourceSet")
+        (project.findProperty("testOutputFramework") as java.io.File).absolutePath
     }
 
     override fun configure(config: Closure<*>): Task {