ParentContext () { PsiElement parent = myExpression.getParent(); if (parent == myLambda) { return myVoidMode ? ParentContext.EXPRESSION : ParentContext.RETURN; } return super.getExpectedParentContext(); }