diff --git a/uast/uast-common/src/org/jetbrains/uast/UastContext.kt b/uast/uast-common/src/org/jetbrains/uast/UastContext.kt
index 0a6205feb42a..108d5fa87ffb 100644
--- a/uast/uast-common/src/org/jetbrains/uast/UastContext.kt
+++ b/uast/uast-common/src/org/jetbrains/uast/UastContext.kt
@@ -42,11 +42,7 @@ class UastContext(val project: Project) : UastLanguagePlugin {
     get() = UastLanguagePlugin.getInstances()
 
   fun findPlugin(element: PsiElement): UastLanguagePlugin? {
-    // we're searching for plugin for file because sometimes java elements are used in another languages (see Drools)
-    val containingFile = element.containingFile?.takeIf {
-      it !is PsiCompiledFile // Don't trust compiled files because of KT-18054
-    }
-    val language = (containingFile ?: element).language
+    val language = element.language
     return languagePlugins.firstOrNull { it.language == language }
   }