From eb73cc587bacd45f0e6806ce76d9de4840346463 Mon Sep 17 00:00:00 2001
From: Yaroslav Pankratyev <yaroslav.pankratyev@jetbrains.com>
Date: Wed, 25 Apr 2018 16:11:35 +0700
Subject: [PATCH] IDEA-159541 More improvements in UnstableApiUsageInspection

---
 .../src/inspections/UnstableApiUsageInspection.java         | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/plugins/devkit/devkit-core/src/inspections/UnstableApiUsageInspection.java b/plugins/devkit/devkit-core/src/inspections/UnstableApiUsageInspection.java
index c951ffa3e198..eecdd335c928 100644
--- a/plugins/devkit/devkit-core/src/inspections/UnstableApiUsageInspection.java
+++ b/plugins/devkit/devkit-core/src/inspections/UnstableApiUsageInspection.java
@@ -98,7 +98,7 @@ public class UnstableApiUsageInspection extends LocalInspectionTool {
     if (containingVirtualFile == null) {
       return false;
     }
-    return ProjectFileIndex.getInstance(element.getProject()).isInLibrary(containingVirtualFile);
+    return ProjectFileIndex.getInstance(element.getProject()).isInLibraryClasses(containingVirtualFile);
   }
 
   @NotNull
@@ -123,9 +123,7 @@ public class UnstableApiUsageInspection extends LocalInspectionTool {
       if (((ResolvingHint)reference).canResolveTo(PsiModifierListOwner.class)) {
         return (PsiModifierListOwner)reference.resolve();
       }
-      else {
-        return null;
-      }
+      return null;
     }
 
     PsiElement resolvedElement = reference.resolve();