PsiElement (PsiMethodCallExpression call) { PsiMethodCallExpression qualifierCall = getQualifierMethodCall(call); if (!STREAM_ITERATE.test(qualifierCall)) return null; CommentTracker ct = new CommentTracker(); PsiExpression predicate = call.getArgumentList().getExpressions()[0]; PsiExpressionList argList = qualifierCall.getArgumentList(); argList.addAfter(ct.markUnchanged(predicate), argList.getExpressions()[0]); return ct.replaceAndRestoreComments(call, qualifierCall); }