From cd481c159bfb6ba77958a1ba1da8dfed805645dd Mon Sep 17 00:00:00 2001
From: Tim te Beek <tim@moderne.io>
Date: Sun, 18 Aug 2024 00:11:53 +0200
Subject: [PATCH] Move `doAfterVisit` to within conditional

---
 .../java/testing/junit5/UpdateTestAnnotation.java           | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/src/main/java/org/openrewrite/java/testing/junit5/UpdateTestAnnotation.java b/src/main/java/org/openrewrite/java/testing/junit5/UpdateTestAnnotation.java
index f7f3314b8..dbe906653 100644
--- a/src/main/java/org/openrewrite/java/testing/junit5/UpdateTestAnnotation.java
+++ b/src/main/java/org/openrewrite/java/testing/junit5/UpdateTestAnnotation.java
@@ -55,7 +55,6 @@ public TreeVisitor<?, ExecutionContext> getVisitor() {
     private static class UpdateTestAnnotationVisitor extends JavaIsoVisitor<ExecutionContext> {
         private static final AnnotationMatcher JUNIT4_TEST = new AnnotationMatcher("@org.junit.Test");
 
-        
         private JavaParser.@Nullable Builder<?, ?> javaParser;
 
         private JavaParser.Builder<?, ?> javaParser(ExecutionContext ctx) {
@@ -66,7 +65,6 @@ private static class UpdateTestAnnotationVisitor extends JavaIsoVisitor<Executio
             return javaParser;
         }
 
-
         @Override
         public J.CompilationUnit visitCompilationUnit(J.CompilationUnit cu, ExecutionContext ctx) {
             J.CompilationUnit c = super.visitCompilationUnit(cu, ctx);
@@ -169,9 +167,9 @@ public J.MethodDeclaration visitMethodDeclaration(J.MethodDeclaration method, Ex
                     maybeAddImport("java.util.concurrent.TimeUnit");
                 }
                 maybeAddImport("org.junit.jupiter.api.Test");
-            }
 
-            doAfterVisit(new LambdaBlockToExpression().getVisitor());
+                doAfterVisit(new LambdaBlockToExpression().getVisitor());
+            }
 
             return super.visitMethodDeclaration(m, ctx);
         }