void (@Nullable PsiElement element, @NotNull UastCodeGenerationPlugin generationPlugin) { UElement uElement = UastContextKt.toUElement(element); if (uElement == null) { return; } // Here we rely on that replacing same element will shorten references generationPlugin.replace(uElement, uElement, UElement.class); }