Use ImportService rather than ShortenFullyQualifiedTypeReferences

(cherry picked from commit f64aea9)
openrewrite · Nov 21, 2023 · e69126c · e69126c
1 parent dc42627
commit e69126c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/main/java/org/openrewrite/java/template/internal/AbstractRefasterJavaVisitor.java b/src/main/java/org/openrewrite/java/template/internal/AbstractRefasterJavaVisitor.java
@@ -19,9 +19,9 @@
 import org.openrewrite.ExecutionContext;
 import org.openrewrite.TreeVisitor;
 import org.openrewrite.java.JavaVisitor;
-import org.openrewrite.java.ShortenFullyQualifiedTypeReferences;
 import org.openrewrite.java.cleanup.SimplifyBooleanExpressionVisitor;
 import org.openrewrite.java.cleanup.UnnecessaryParenthesesVisitor;
+import org.openrewrite.java.service.ImportService;
 import org.openrewrite.java.tree.J;
 
 import java.util.EnumSet;
@@ -42,7 +42,7 @@ protected J embed(J j, Cursor cursor, ExecutionContext ctx, EmbeddingOption... o
             j = new MinimumViableParentheses().visitNonNull(j, ctx, cursor.getParentOrThrow());
         }
         if (optionsSet.contains(EmbeddingOption.SHORTEN_NAMES)) {
-            doAfterVisit(ShortenFullyQualifiedTypeReferences.modifyOnly(j));
+            doAfterVisit(service(ImportService.class).shortenFullyQualifiedTypeReferencesIn(j));
         }
         if (optionsSet.contains(EmbeddingOption.SIMPLIFY_BOOLEANS)) {
             j = new SimplifyBooleanExpressionVisitor().visitNonNull(j, ctx, cursor.getParentOrThrow());