Update src/main/java/org/openrewrite/kotlin/tree/K.java

Co-authored-by: Sam Snyder <[email protected]>
openrewrite · Apr 14, 2023 · 1b3b8dc · 1b3b8dc
1 parent 3b18f9a
commit 1b3b8dc
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/main/java/org/openrewrite/kotlin/tree/K.java b/src/main/java/org/openrewrite/kotlin/tree/K.java
@@ -480,10 +480,10 @@ public <J2 extends Tree> J2 withMarkers(Markers markers) {
 
         @Override
         public <J2 extends J> J2 withType(@Nullable JavaType type) {
-            if (typedTree instanceof J.NewClass) {
-                return (J2) this;
+            if (typedTree instanceof FunctionType) {
+                return (J2) withTypedTree(typedTree.withType(type));
             }
-            return (J2) withTypedTree(typedTree.withType(type));
+            return (J2) this;
         }
 
         @Override