From b237afdddd09af442b30f2fcb012bf0f90f8dfe4 Mon Sep 17 00:00:00 2001
From: Peter Gromov <peter@jetbrains.com>
Date: Fri, 20 Dec 2019 09:43:02 +0100
Subject: [PATCH] AbstractBatchSuppressByNoInspectionCommentFix: fix EA-220060
 - PIEAE

GitOrigin-RevId: a77d2a2c2bc00d9f9bb17882c38f26e5be4f0b37
---
 .../AbstractBatchSuppressByNoInspectionCommentFix.java        | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/platform/analysis-impl/src/com/intellij/codeInsight/daemon/impl/actions/AbstractBatchSuppressByNoInspectionCommentFix.java b/platform/analysis-impl/src/com/intellij/codeInsight/daemon/impl/actions/AbstractBatchSuppressByNoInspectionCommentFix.java
index d8c559b3208e..0ea81eccd6fb 100644
--- a/platform/analysis-impl/src/com/intellij/codeInsight/daemon/impl/actions/AbstractBatchSuppressByNoInspectionCommentFix.java
+++ b/platform/analysis-impl/src/com/intellij/codeInsight/daemon/impl/actions/AbstractBatchSuppressByNoInspectionCommentFix.java
@@ -12,6 +12,7 @@ import com.intellij.openapi.util.Iconable;
 import com.intellij.openapi.util.text.StringUtil;
 import com.intellij.psi.PsiComment;
 import com.intellij.psi.PsiElement;
+import com.intellij.psi.PsiFile;
 import com.intellij.psi.util.PsiTreeUtil;
 import com.intellij.util.IncorrectOperationException;
 import com.intellij.util.ThreeState;
@@ -126,12 +127,13 @@ public abstract class AbstractBatchSuppressByNoInspectionCommentFix implements C
   public void invoke(@NotNull final Project project, @NotNull final PsiElement element) throws IncorrectOperationException {
     if (!isAvailable(project, element)) return;
     PsiElement container = getContainer(element);
+    PsiFile file = element.getContainingFile();
     if (container == null) return;
 
     if (replaceSuppressionComments(container)) return;
 
     createSuppression(project, element, container);
-    UndoUtil.markPsiFileForUndo(element.getContainingFile());
+    UndoUtil.markPsiFileForUndo(file);
   }
 
   protected boolean replaceSuppressionComments(PsiElement container) {