diff --git a/contrib/spring-ai/src/test/java/com/google/adk/models/springai/SpringAIIntegrationTest.java b/contrib/spring-ai/src/test/java/com/google/adk/models/springai/SpringAIIntegrationTest.java
index 6843c8eaa..e6e0b7e77 100644
--- a/contrib/spring-ai/src/test/java/com/google/adk/models/springai/SpringAIIntegrationTest.java
+++ b/contrib/spring-ai/src/test/java/com/google/adk/models/springai/SpringAIIntegrationTest.java
@@ -80,7 +80,11 @@ public ChatResponse call(Prompt prompt) {
 
     List<Event> events =
         runner
-            .runAsync(session, userMessage, com.google.adk.agents.RunConfig.builder().build())
+            .runAsync(
+                session.userId(),
+                session.id(),
+                userMessage,
+                com.google.adk.agents.RunConfig.builder().build())
             .toList()
             .blockingGet();
 
@@ -159,7 +163,11 @@ public ChatResponse call(Prompt prompt) {
 
     List<Event> events =
         runner
-            .runAsync(session, userMessage, com.google.adk.agents.RunConfig.builder().build())
+            .runAsync(
+                session.userId(),
+                session.id(),
+                userMessage,
+                com.google.adk.agents.RunConfig.builder().build())
             .toList()
             .blockingGet();
 
@@ -228,7 +236,8 @@ public Flux<ChatResponse> stream(Prompt prompt) {
     List<Event> events =
         runner
             .runAsync(
-                session,
+                session.userId(),
+                session.id(),
                 userMessage,
                 com.google.adk.agents.RunConfig.builder()
                     .setStreamingMode(com.google.adk.agents.RunConfig.StreamingMode.SSE)
diff --git a/contrib/spring-ai/src/test/java/com/google/adk/models/springai/TestUtils.java b/contrib/spring-ai/src/test/java/com/google/adk/models/springai/TestUtils.java
index f18ded055..2f39686d7 100644
--- a/contrib/spring-ai/src/test/java/com/google/adk/models/springai/TestUtils.java
+++ b/contrib/spring-ai/src/test/java/com/google/adk/models/springai/TestUtils.java
@@ -46,7 +46,8 @@ public static List<Event> askAgent(BaseAgent agent, boolean streaming, Object...
       allEvents.addAll(
           runner
               .runAsync(
-                  session,
+                  session.userId(),
+                  session.id(),
                   messageContent,
                   RunConfig.builder()
                       .setStreamingMode(
@@ -73,7 +74,10 @@ public static List<Event> askBlockingAgent(BaseAgent agent, Object... messages)
 
     for (Content content : contents) {
       List<Event> batchEvents =
-          runner.runAsync(session, content, RunConfig.builder().build()).toList().blockingGet();
+          runner
+              .runAsync(session.userId(), session.id(), content, RunConfig.builder().build())
+              .toList()
+              .blockingGet();
       events.addAll(batchEvents);
     }
 
@@ -96,7 +100,8 @@ public static List<Event> askAgentStreaming(BaseAgent agent, Object... messages)
       List<Event> batchEvents =
           runner
               .runAsync(
-                  session,
+                  session.userId(),
+                  session.id(),
                   content,
                   RunConfig.builder().setStreamingMode(RunConfig.StreamingMode.SSE).build())
               .toList()