SimpleTraitMatcher navigate subtrees in asVisit on matching traits

openrewrite · Jul 9, 2024 · 03aa622 · 03aa622
1 parent 1c11e1f
commit 03aa622
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/rewrite-core/src/main/java/org/openrewrite/trait/SimpleTraitMatcher.java b/rewrite-core/src/main/java/org/openrewrite/trait/SimpleTraitMatcher.java
@@ -76,7 +76,7 @@ public <P> TreeVisitor<? extends Tree, P> asVisitor(VisitFunction2<U, P> visitor
             public @Nullable Tree preVisit(Tree tree, P p) {
                 U u = test(getCursor());
                 return u != null ?
-                        visitor.visit(u, p) :
+                        super.visit(visitor.visit(u, p), p) :
                         super.visit(tree, p);
             }
         };