Fix compilation on K.CompilationUnit.service (#593)

After openrewrite/rewrite@33d8f7d
openrewrite · Feb 6, 2024 · bce277f · bce277f
1 parent 74384ac
commit bce277f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/main/java/org/openrewrite/kotlin/tree/K.java b/src/main/java/org/openrewrite/kotlin/tree/K.java
@@ -341,7 +341,7 @@ public K.CompilationUnit withStatements(List<JRightPadded<Statement>> statements
 
         @Override
         @SuppressWarnings("unchecked")
-        public <S> S service(Class<S> service) {
+        public <S, T extends S> T service(Class<S> service) {
             String serviceName = service.getName();
             try {
                 Class<S> serviceClass;
@@ -356,7 +356,7 @@ public <S> S service(Class<S> service) {
                 } else {
                     return JavaSourceFile.super.service(service);
                 }
-                return serviceClass.getConstructor().newInstance();
+                return (T) serviceClass.getConstructor().newInstance();
             } catch (Exception e) {
                 throw new RuntimeException(e);
             }