diff --git a/src/main/java/org/openrewrite/kotlin/tree/K.java b/src/main/java/org/openrewrite/kotlin/tree/K.java
index c73547768..a21492875 100644
--- 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