@Override
    public void visitYieldStatement(@NotNull PsiYieldStatement statement) {
      PsiSwitchExpression expression = statement.findEnclosingExpression();
      if (expression != null) {
        expectedType = expression.getType();
      }
    }
@Override
    public void visitYieldStatement(@NotNull PsiYieldStatement statement)