diff --git a/uast/uast-common/src/org/jetbrains/uast/UastContext.kt b/uast/uast-common/src/org/jetbrains/uast/UastContext.kt
index e59220ce4762..d999879eb0d2 100644
--- a/uast/uast-common/src/org/jetbrains/uast/UastContext.kt
+++ b/uast/uast-common/src/org/jetbrains/uast/UastContext.kt
@@ -42,7 +42,11 @@ class UastContext(val project: Project) : UastLanguagePlugin {
     get() = UastLanguagePlugin.getInstances()
 
   fun findPlugin(element: PsiElement): UastLanguagePlugin? {
-    val language = element.language
+    // 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
     return languagePlugins.firstOrNull { it.language == language }
   }