PsiExpression (@NotNull PsiExpression expression) { PsiMethodCallExpression methodCall = tryCast(expression.getParent(), PsiMethodCallExpression.class); if (methodCall != null && methodCall.getMethodExpression() == expression) { return methodCall; } return expression; }