From bce277fa71456b417562e4b705a705ae42d33248 Mon Sep 17 00:00:00 2001
From: Tim te Beek <tim@moderne.io>
Date: Tue, 6 Feb 2024 18:32:58 +0100
Subject: [PATCH] Fix compilation on K.CompilationUnit.service (#593)

After https://github.com/openrewrite/rewrite/commit/33d8f7d42f9bd6749ab93171fad31289b0e2bc97
---
 src/main/java/org/openrewrite/kotlin/tree/K.java | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/main/java/org/openrewrite/kotlin/tree/K.java b/src/main/java/org/openrewrite/kotlin/tree/K.java
index 302e9931..494b4dc2 100644
--- 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);
             }