diff --git a/rewrite-java/src/main/java/org/openrewrite/java/ReplaceAnnotation.java b/rewrite-java/src/main/java/org/openrewrite/java/ReplaceAnnotation.java index a011406b816..03b8e237c1f 100644 --- a/rewrite-java/src/main/java/org/openrewrite/java/ReplaceAnnotation.java +++ b/rewrite-java/src/main/java/org/openrewrite/java/ReplaceAnnotation.java @@ -19,6 +19,7 @@ import lombok.Value; import org.openrewrite.*; import org.openrewrite.internal.lang.Nullable; +import org.openrewrite.java.service.ImportService; import org.openrewrite.java.tree.J; import org.openrewrite.java.tree.JavaCoordinates; import org.openrewrite.java.tree.TypeUtils; @@ -91,7 +92,7 @@ public J.Annotation visitAnnotation(J.Annotation annotation, ExecutionContext ct maybeRemoveImport(TypeUtils.asFullyQualified(a.getType())); JavaCoordinates replaceCoordinate = a.getCoordinates().replace(); a = replacement.apply(getCursor(), replaceCoordinate); - doAfterVisit(ShortenFullyQualifiedTypeReferences.modifyOnly(a)); + doAfterVisit(service(ImportService.class).shortenFullyQualifiedTypeReferencesIn(a)); return a; } }