Minor: Cleanup jps.gradle.kts

gradle/jps.gradle.kts

diff --git a/gradle/jps.gradle.kts b/gradle/jps.gradle.kts
index 3d4f351..edb31f6 100644
--- a/gradle/jps.gradle.kts
+++ b/gradle/jps.gradle.kts
@@ -1,6 +1,8 @@
+@file:Suppress("UnstableApiUsage")
+
 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
@@ -11,7 +13,7 @@
 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",
@@ -129,7 +131,7 @@
                             mainClass = "org.jetbrains.kotlin.pill.generateAllTests.Main"
                         }
 
-                        defaults<org.jetbrains.gradle.ext.JUnit> {
+                        defaults<JUnit> {
                             configureForKotlin()
                         }