diff --git a/java/java-impl/src/com/intellij/codeInsight/generation/ui/GenerateEqualsWizard.java b/java/java-impl/src/com/intellij/codeInsight/generation/ui/GenerateEqualsWizard.java index d3e20aa67ff6..a9e33e0d7468 100644 --- a/java/java-impl/src/com/intellij/codeInsight/generation/ui/GenerateEqualsWizard.java +++ b/java/java-impl/src/com/intellij/codeInsight/generation/ui/GenerateEqualsWizard.java @@ -154,8 +154,7 @@ public class GenerateEqualsWizard extends AbstractGenerateEqualsWizard list = new ArrayList<>(); for (MemberInfo equalsMemberInfo : equalsMemberInfos) { - PsiField field = (PsiField)equalsMemberInfo.getMember(); - if (!(field.getType() instanceof PsiPrimitiveType)) { + if (mayNeedNullCheck((PsiField)equalsMemberInfo.getMember())) { list.add(myFieldsToNonNull.get(equalsMemberInfo.getMember())); } } @@ -227,11 +226,8 @@ public class GenerateEqualsWizard extends AbstractGenerateEqualsWizard