diff --git a/src/test/java/org/antlr/intellij/plugin/folding/ANTLRv4FoldingBuilderTest.java b/src/test/java/org/antlr/intellij/plugin/folding/ANTLRv4FoldingBuilderTest.java
index 61e7fd78..c82a6225 100644
--- a/src/test/java/org/antlr/intellij/plugin/folding/ANTLRv4FoldingBuilderTest.java
+++ b/src/test/java/org/antlr/intellij/plugin/folding/ANTLRv4FoldingBuilderTest.java
@@ -4,12 +4,12 @@
 import com.intellij.openapi.editor.Editor;
 import com.intellij.openapi.editor.FoldRegion;
 import com.intellij.testFramework.EditorTestUtil;
-import com.intellij.testFramework.fixtures.LightPlatformCodeInsightFixtureTestCase;
+import com.intellij.testFramework.fixtures.BasePlatformTestCase;
 import org.antlr.intellij.plugin.TestUtils;
 
 import java.lang.reflect.Method;
 
-public class ANTLRv4FoldingBuilderTest extends LightPlatformCodeInsightFixtureTestCase {
+public class ANTLRv4FoldingBuilderTest extends BasePlatformTestCase {
 
 	public void test_folding_should_not_throw_on_incomplete_prequel() {
 		// Given
@@ -48,4 +48,4 @@ private void buildInitialFoldings() {
 			CodeFoldingManager.getInstance(getProject()).buildInitialFoldings(myFixture.getEditor());
 		}
 	}
-}
\ No newline at end of file
+}
diff --git a/src/test/java/org/antlr/intellij/plugin/psi/GrammarElementRefTest.java b/src/test/java/org/antlr/intellij/plugin/psi/GrammarElementRefTest.java
index 141546c6..679ef45d 100644
--- a/src/test/java/org/antlr/intellij/plugin/psi/GrammarElementRefTest.java
+++ b/src/test/java/org/antlr/intellij/plugin/psi/GrammarElementRefTest.java
@@ -3,7 +3,7 @@
 import com.intellij.openapi.editor.markup.RangeHighlighter;
 import com.intellij.psi.PsiElement;
 import com.intellij.psi.PsiReference;
-import com.intellij.testFramework.fixtures.LightPlatformCodeInsightFixtureTestCase;
+import com.intellij.testFramework.fixtures.BasePlatformTestCase;
 import com.intellij.usageView.UsageInfo;
 import org.antlr.intellij.plugin.ANTLRv4FileRoot;
 import org.antlr.intellij.plugin.ANTLRv4PluginController;
@@ -13,7 +13,7 @@
 import java.util.Collection;
 import java.util.function.Consumer;
 
-public class GrammarElementRefTest extends LightPlatformCodeInsightFixtureTestCase {
+public class GrammarElementRefTest extends BasePlatformTestCase {
 	public void testFindUsagesOfLexerRule() {
 		Collection<UsageInfo> ruleUsages = myFixture.testFindUsages("SimpleGrammar.g4");
 		assertEquals(3, ruleUsages.size());
@@ -249,4 +249,4 @@ private PsiElement resolveRefAtCaret() {
 		return null;
 	}
 
-}
\ No newline at end of file
+}