Fix TabsAndIndentsVisitor after adding J.Modifier to visitor

openrewrite · Oct 15, 2024 · c2cb011 · c2cb011
1 parent 2f1b957
commit c2cb011
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/main/java/org/openrewrite/kotlin/format/TabsAndIndentsVisitor.java b/src/main/java/org/openrewrite/kotlin/format/TabsAndIndentsVisitor.java
@@ -171,8 +171,8 @@ public Space visitSpace(Space space, Space.Location loc, P p) {
                 }
             }
 
-            if (loc == Space.Location.MODIFIER_PREFIX && value instanceof J.MethodDeclaration) {
-                J.MethodDeclaration m = getCursor().getValue();
+            if (loc == Space.Location.MODIFIER_PREFIX && value instanceof J.Modifier && parent.getValue() instanceof J.MethodDeclaration) {
+                J.MethodDeclaration m = parent.getValue();
                 if (!m.getLeadingAnnotations().isEmpty()) {
                     alignToAnnotation = true;
                 }