diff --git a/idea/tests/org/jetbrains/kotlin/formatter/AbstractFormatterTest.java b/idea/tests/org/jetbrains/kotlin/formatter/AbstractFormatterTest.java
index cd5722eacf5..6bf6240e5e5 100644
--- a/idea/tests/org/jetbrains/kotlin/formatter/AbstractFormatterTest.java
+++ b/idea/tests/org/jetbrains/kotlin/formatter/AbstractFormatterTest.java
@@ -17,11 +17,11 @@ import com.intellij.psi.PsiDocumentManager;
 import com.intellij.psi.PsiFile;
 import com.intellij.psi.codeStyle.CodeStyleManager;
 import com.intellij.psi.codeStyle.CodeStyleSettings;
-import com.intellij.testFramework.LightIdeaTestCase;
 import com.intellij.util.IncorrectOperationException;
 import org.jetbrains.annotations.NonNls;
 import org.jetbrains.annotations.NotNull;
 import org.jetbrains.kotlin.idea.KotlinLanguage;
+import org.jetbrains.kotlin.idea.test.KotlinLightIdeaTestCase;
 import org.jetbrains.kotlin.idea.test.PluginTestCaseBase;
 import org.jetbrains.kotlin.test.InTextDirectivesUtils;
 import org.jetbrains.kotlin.test.KotlinTestUtils;
@@ -33,7 +33,7 @@ import java.util.Map;
 
 // Based on from com.intellij.psi.formatter.java.AbstractJavaFormatterTest
 @SuppressWarnings("UnusedDeclaration")
-public abstract class AbstractFormatterTest extends LightIdeaTestCase {
+public abstract class AbstractFormatterTest extends KotlinLightIdeaTestCase {
 
     protected enum Action {REFORMAT, INDENT}
 
diff --git a/idea/tests/org/jetbrains/kotlin/formatter/AbstractFormatterTest.java.191 b/idea/tests/org/jetbrains/kotlin/formatter/AbstractFormatterTest.java.191
index e8aa2e78359..56c2a3ca83a 100644
--- a/idea/tests/org/jetbrains/kotlin/formatter/AbstractFormatterTest.java.191
+++ b/idea/tests/org/jetbrains/kotlin/formatter/AbstractFormatterTest.java.191
@@ -17,11 +17,11 @@ import com.intellij.psi.PsiDocumentManager;
 import com.intellij.psi.PsiFile;
 import com.intellij.psi.codeStyle.CodeStyleManager;
 import com.intellij.psi.codeStyle.CodeStyleSettings;
-import com.intellij.testFramework.LightIdeaTestCase;
 import com.intellij.util.IncorrectOperationException;
 import org.jetbrains.annotations.NonNls;
 import org.jetbrains.annotations.NotNull;
 import org.jetbrains.kotlin.idea.KotlinLanguage;
+import org.jetbrains.kotlin.idea.test.KotlinLightIdeaTestCase;
 import org.jetbrains.kotlin.idea.test.PluginTestCaseBase;
 import org.jetbrains.kotlin.test.InTextDirectivesUtils;
 import org.jetbrains.kotlin.test.KotlinTestUtils;
@@ -33,7 +33,7 @@ import java.util.Map;
 
 // Based on from com.intellij.psi.formatter.java.AbstractJavaFormatterTest
 @SuppressWarnings("UnusedDeclaration")
-public abstract class AbstractFormatterTest extends LightIdeaTestCase {
+public abstract class AbstractFormatterTest extends KotlinLightIdeaTestCase {
 
     protected enum Action {REFORMAT, INDENT}
 
diff --git a/idea/tests/org/jetbrains/kotlin/formatter/AbstractFormatterTest.java.193 b/idea/tests/org/jetbrains/kotlin/formatter/AbstractFormatterTest.java.193
index 3aeee21e14d..bbd9aa8cbca 100644
--- a/idea/tests/org/jetbrains/kotlin/formatter/AbstractFormatterTest.java.193
+++ b/idea/tests/org/jetbrains/kotlin/formatter/AbstractFormatterTest.java.193
@@ -17,11 +17,11 @@ import com.intellij.psi.PsiDocumentManager;
 import com.intellij.psi.PsiFile;
 import com.intellij.psi.codeStyle.CodeStyleManager;
 import com.intellij.psi.codeStyle.CodeStyleSettings;
-import com.intellij.testFramework.LightIdeaTestCase;
 import com.intellij.util.IncorrectOperationException;
 import org.jetbrains.annotations.NonNls;
 import org.jetbrains.annotations.NotNull;
 import org.jetbrains.kotlin.idea.KotlinLanguage;
+import org.jetbrains.kotlin.idea.test.KotlinLightIdeaTestCase;
 import org.jetbrains.kotlin.idea.test.PluginTestCaseBase;
 import org.jetbrains.kotlin.test.InTextDirectivesUtils;
 import org.jetbrains.kotlin.test.KotlinTestUtils;
@@ -33,7 +33,7 @@ import java.util.Map;
 
 // Based on from com.intellij.psi.formatter.java.AbstractJavaFormatterTest
 @SuppressWarnings("UnusedDeclaration")
-public abstract class AbstractFormatterTest extends LightIdeaTestCase {
+public abstract class AbstractFormatterTest extends KotlinLightIdeaTestCase {
 
     protected enum Action {REFORMAT, INDENT}
 
diff --git a/idea/tests/org/jetbrains/kotlin/idea/framework/ui/LoadVersionsFromMavenTest.java b/idea/tests/org/jetbrains/kotlin/idea/framework/ui/LoadVersionsFromMavenTest.java
index a959a8e67be..967567a3e10 100644
--- a/idea/tests/org/jetbrains/kotlin/idea/framework/ui/LoadVersionsFromMavenTest.java
+++ b/idea/tests/org/jetbrains/kotlin/idea/framework/ui/LoadVersionsFromMavenTest.java
@@ -5,14 +5,15 @@
 
 package org.jetbrains.kotlin.idea.framework.ui;
 
-import com.intellij.testFramework.LightIdeaTestCase;
 import com.intellij.util.text.VersionComparatorUtil;
+import org.jetbrains.kotlin.idea.test.KotlinLightIdeaTestCase;
 import org.jetbrains.kotlin.test.JUnit3WithIdeaConfigurationRunner;
 import org.junit.runner.RunWith;
 
 import java.util.Collection;
+
 @RunWith(JUnit3WithIdeaConfigurationRunner.class)
-public class LoadVersionsFromMavenTest extends LightIdeaTestCase {
+public class LoadVersionsFromMavenTest extends KotlinLightIdeaTestCase {
     public void testDownload() throws Exception {
         Collection<String> versions = ConfigureDialogWithModulesAndVersion.loadVersions("1.0.0");
         assertTrue(versions.size() > 0);