diff --git a/plugins/devkit/intellij.kotlin.devkit/src/driver/PublicTestExtensionFunctionInspection.kt b/plugins/devkit/intellij.kotlin.devkit/src/driver/PublicTestExtensionFunctionInspection.kt
index fdfaab1e8d77..695c3131d231 100644
--- a/plugins/devkit/intellij.kotlin.devkit/src/driver/PublicTestExtensionFunctionInspection.kt
+++ b/plugins/devkit/intellij.kotlin.devkit/src/driver/PublicTestExtensionFunctionInspection.kt
@@ -51,7 +51,8 @@ class PublicTestExtensionFunctionInspection : LocalInspectionTool() {
     if (function.isLocal) return false
 
     //optimization to avoid resolve
-    if (receiverTypeRef.text != "Driver" && receiverTypeRef.text != "Finder" && receiverTypeRef.text != "IdeaFrameUI") return false
+    if (receiverTypeRef.text != "Driver" && receiverTypeRef.text != "Finder" && receiverTypeRef.text != "IdeaFrameUI"
+        && receiverTypeRef.text != "UIComponent") return false
 
     // Check if it's public (no visibility modifier means public in Kotlin)
     val visibilityModifier = function.visibilityModifier()?.text