void (@NotNull PsiYieldStatement statement) { PsiSwitchExpression expression = statement.findEnclosingExpression(); if (expression != null) { Collections.addAll(myResult, getExpectedTypes(expression, myForCompletion)); } }