diff --git a/plugins/InspectionGadgets/test/com/siyeh/igtest/security/design_for_extension/DesignForExtension.java b/plugins/InspectionGadgets/test/com/siyeh/igtest/security/design_for_extension/DesignForExtension.java
index c6639e9758e9..dab95763a6c4 100644
--- a/plugins/InspectionGadgets/test/com/siyeh/igtest/security/design_for_extension/DesignForExtension.java
+++ b/plugins/InspectionGadgets/test/com/siyeh/igtest/security/design_for_extension/DesignForExtension.java
@@ -3,7 +3,7 @@ package com.siyeh.igtest.security.design_for_extension;
 class DesignForExtension {
 
 
-  public boolean equals(Object o) {
+  public boolean <warning descr="Method 'equals()' may be overridden and its functionality ignored">equals</warning>(Object o) {
     return true;
   }
 
diff --git a/plugins/InspectionGadgets/test/com/siyeh/igtest/security/design_for_extension/expected.xml b/plugins/InspectionGadgets/test/com/siyeh/igtest/security/design_for_extension/expected.xml
deleted file mode 100644
index d1054673ef8b..000000000000
--- a/plugins/InspectionGadgets/test/com/siyeh/igtest/security/design_for_extension/expected.xml
+++ /dev/null
@@ -1,9 +0,0 @@
-<?xml version="1.0" encoding="UTF-8"?>
-<problems>
-  <problem>
-    <file>DesignForExtension.java</file>
-    <line>6</line>
-    <problem_class severity="WARNING" attribute_key="WARNING_ATTRIBUTES">Design for extension</problem_class>
-    <description>Method &lt;code&gt;equals()&lt;/code&gt; may be overridden and its functionality ignored #loc</description>
-  </problem>
-</problems>
\ No newline at end of file
diff --git a/plugins/InspectionGadgets/testsrc/com/siyeh/ig/security/DesignForExtensionInspectionTest.java b/plugins/InspectionGadgets/testsrc/com/siyeh/ig/security/DesignForExtensionInspectionTest.java
index 590cdac89b11..689be916649b 100644
--- a/plugins/InspectionGadgets/testsrc/com/siyeh/ig/security/DesignForExtensionInspectionTest.java
+++ b/plugins/InspectionGadgets/testsrc/com/siyeh/ig/security/DesignForExtensionInspectionTest.java
@@ -1,10 +1,16 @@
 package com.siyeh.ig.security;
 
-import com.siyeh.ig.IGInspectionTestCase;
+import com.intellij.codeInspection.InspectionProfileEntry;
+import com.siyeh.ig.LightInspectionTestCase;
+import org.jetbrains.annotations.Nullable;
 
-public class DesignForExtensionInspectionTest extends IGInspectionTestCase {
+public class DesignForExtensionInspectionTest extends LightInspectionTestCase {
 
-  public void test() throws Exception {
-    doTest("com/siyeh/igtest/security/design_for_extension", new DesignForExtensionInspection());
+  public void testDesignForExtension() { doTest(); }
+
+  @Nullable
+  @Override
+  protected InspectionProfileEntry getInspection() {
+    return new DesignForExtensionInspection();
   }
 }
\ No newline at end of file