void (@NotNull PsiExpressionStatement statement) { super.visitExpressionStatement(statement); PsiElement parent = statement.getParent(); if (parent instanceof PsiSwitchLabeledRuleStatement) { PsiSwitchBlock switchBlock = ((PsiSwitchLabeledRuleStatement)parent).getEnclosingSwitchBlock(); if (switchBlock instanceof PsiSwitchExpression && !PsiPolyExpressionUtil.isPolyExpression((PsiExpression)switchBlock)) { add(HighlightUtil.checkYieldExpressionType(statement.getExpression())); } } }