TailType (@NotNull PsiAssignmentExpression assignment) { if (assignment.getParent() instanceof PsiExpressionStatement) { if (!(assignment.getParent().getParent() instanceof PsiForStatement forStatement)) { return TailTypes.semicolonType(); } if (!assignment.getParent().equals(forStatement.getUpdate())) { return TailTypes.semicolonType(); } } return TailTypes.noneType(); }