void (@NotNull PsiMethodCallExpression expression) { super.visitMethodCallExpression(expression); PsiMethod method = expression.resolveMethod(); if (method != null && generatedClass.equals(method.getContainingClass())) { List<PsiMethod> methods = ContainerUtil.filter(extractedClass.getMethods(), x -> isSameMethod(x, method)); if (methods.size() == 1) { LOG.info("Replace method: " + method.getName()); method.putUserData(LightMethodObjectExtractedData.REFERENCE_METHOD, methods.get(0)); } } }