void (@NotNull Project project, @NotNull PsiElement element, @NotNull ModPsiUpdater updater) { PsiBlockStatement body = PsiTreeUtil.getParentOfType(element, PsiBlockStatement.class); if (body != null && body.getParent() instanceof PsiSwitchLabeledRuleStatement) { PsiStatement bodyStatement = getSingleStatement(body.getCodeBlock()); if (bodyStatement instanceof PsiYieldStatement) { unwrapYieldValue(body, (PsiYieldStatement)bodyStatement); } else if (bodyStatement instanceof PsiThrowStatement || bodyStatement instanceof PsiExpressionStatement) { unwrap(body, bodyStatement); } } }