void (@NotNull String varName) { PsiStatement statement = addStatement(String.format("sink(%s);", varName)); PsiMethodCallExpression call = PsiTreeUtil.findChildOfType(statement, PsiMethodCallExpression.class); mySinkCallArgument = call.getArgumentList().getExpressions()[0]; }