diff --git a/build.gradle b/build.gradle
index 28d98c7..c9df5a4 100644
--- a/build.gradle
+++ b/build.gradle
@@ -40,9 +40,9 @@ task generateParser(type: GenerateParser) {
 
 // See https://github.com/JetBrains/gradle-intellij-plugin/
 intellij {
-    // Test with 2020.3
-    version '2020.3'
-    plugins = ['PsiViewer:203-SNAPSHOT', 'org.rust.lang:0.3.136.3514-203', 'org.toml.lang:0.2.136.3514-203']
+    // Test with the latest EAP
+    version 'LATEST-EAP-SNAPSHOT'
+    plugins = ['PsiViewer:211-SNAPSHOT', 'org.rust.lang:0.3.140.3644-203', 'org.toml.lang:0.2.140.3644-203']
 }
 compileKotlin {
     kotlinOptions {
diff --git a/src/main/kotlin/com/mdrobnak/lalrpop/psi/LpResolveType.kt b/src/main/kotlin/com/mdrobnak/lalrpop/psi/LpResolveType.kt
index c71ca66..f9446de 100644
--- a/src/main/kotlin/com/mdrobnak/lalrpop/psi/LpResolveType.kt
+++ b/src/main/kotlin/com/mdrobnak/lalrpop/psi/LpResolveType.kt
@@ -4,7 +4,7 @@ import com.intellij.openapi.project.Project
 import com.intellij.psi.PsiElement
 import com.intellij.psi.PsiFile
 import com.intellij.psi.PsiFileFactory
-import com.intellij.psi.util.findDescendantOfType
+import com.intellij.psi.util.PsiTreeUtil
 import com.mdrobnak.lalrpop.psi.ext.rustGenericUnitStructs
 import com.mdrobnak.lalrpop.psi.util.lalrpopTypeResolutionContext
 import org.rust.lang.RsLanguage
@@ -116,7 +116,7 @@ fun String.lalrpopRustType(
     for (child in file.childrenOfType<RsExpandedElement>())
         child.setContext(modDefinition)
 
-    val tyAlias = file.findDescendantOfType<RsTypeAlias>() ?: return null
+    val tyAlias = PsiTreeUtil.findChildOfType(file, RsTypeAlias::class.java) ?: return null
     val ty = tyAlias.typeReference?.type ?: return null
     return ImplLookup.relativeTo(tyAlias).ctx.fullyResolve(ty)
 }
\ No newline at end of file
diff --git a/src/main/kotlin/com/mdrobnak/lalrpop/psi/ext/LpGrammarDecl.kt b/src/main/kotlin/com/mdrobnak/lalrpop/psi/ext/LpGrammarDecl.kt
index 3055b88..9c64506 100644
--- a/src/main/kotlin/com/mdrobnak/lalrpop/psi/ext/LpGrammarDecl.kt
+++ b/src/main/kotlin/com/mdrobnak/lalrpop/psi/ext/LpGrammarDecl.kt
@@ -2,10 +2,9 @@ package com.mdrobnak.lalrpop.psi.ext
 
 import com.intellij.psi.PsiFile
 import com.intellij.psi.util.PsiTreeUtil
-import com.intellij.psi.util.findDescendantOfType
 import com.mdrobnak.lalrpop.psi.LpGrammarDecl
 
-fun PsiFile.lalrpopFindGrammarDecl(): LpGrammarDecl = findDescendantOfType()!!
+fun PsiFile.lalrpopFindGrammarDecl(): LpGrammarDecl = PsiTreeUtil.findChildOfType(this, LpGrammarDecl::class.java)!!
 
 fun LpGrammarDecl.typeParamsRustUnitStructs(): String =
     grammarTypeParams?.typeParamList?.filter { it.id != null }