diff --git a/gradle/jps.gradle.kts b/gradle/jps.gradle.kts
index 636162f8774..7a2dd7efb91 100644
--- a/gradle/jps.gradle.kts
+++ b/gradle/jps.gradle.kts
@@ -1,7 +1,9 @@
+@file:Suppress("UnstableApiUsage")
+
 import org.gradle.api.internal.project.ProjectInternal
 import org.jetbrains.gradle.ext.*
 import org.jetbrains.kotlin.ideaExt.*
-import org.jetbrains.kotlin.buildUtils.idea.*
+
 
 val ideaPluginDir: File by extra
 val ideaSandboxDir: File by extra
@@ -12,7 +14,7 @@ val intellijUltimateEnabled: Boolean by rootProject.extra
 val ideaUltimatePluginDir: File by rootProject.extra
 val ideaUltimateSandboxDir: File by rootProject.extra
 
-fun org.jetbrains.gradle.ext.JUnit.configureForKotlin() {
+fun JUnit.configureForKotlin() {
     vmParameters = listOf(
         "-ea",
         "-XX:+HeapDumpOnOutOfMemoryError",
@@ -130,7 +132,7 @@ if (kotlinBuildProperties.isInJpsBuildIdeaSync) {
                             mainClass = "org.jetbrains.kotlin.pill.generateAllTests.Main"
                         }
 
-                        defaults<org.jetbrains.gradle.ext.JUnit> {
+                        defaults<JUnit> {
                             configureForKotlin()
                         }