diff --git a/pom.xml b/pom.xml
index c307d2f..c6a9e0b 100644
--- a/pom.xml
+++ b/pom.xml
@@ -4,7 +4,7 @@
   <parent>
     <groupId>io.akka</groupId>
     <artifactId>akka-javasdk-parent</artifactId>
-    <version>3.5.0</version>
+    <version>3.5.1</version>
   </parent>
   
   <groupId>io.akka.ai</groupId>
diff --git a/src/main/java/demo/multiagent/application/EvaluatorAgent.java b/src/main/java/demo/multiagent/application/EvaluatorAgent.java
index c1ecd16..d601fa8 100644
--- a/src/main/java/demo/multiagent/application/EvaluatorAgent.java
+++ b/src/main/java/demo/multiagent/application/EvaluatorAgent.java
@@ -14,8 +14,7 @@
   An agent that acts as an LLM judge to evaluate the quality of AI responses.
   It assesses whether the final answer is appropriate for the original question
   and checks for any deviations from user preferences.
-  """,
-  role = "worker"
+  """
 )
 public class EvaluatorAgent extends Agent {
 
diff --git a/src/main/resources/application.conf b/src/main/resources/application.conf
index c683051..b831764 100644
--- a/src/main/resources/application.conf
+++ b/src/main/resources/application.conf
@@ -3,11 +3,17 @@ akka.javasdk {
     # Other AI models can be configured, see https://doc.akka.io/java/agents.html#model
     # and https://doc.akka.io/java/model-provider-details.html for the reference configurations.
     model-provider = openai
+    # model-provider = googleai-gemini
 
     openai {
       model-name = "gpt-4o-mini"
       # Environment variable override for the API key
       api-key = ${?OPENAI_API_KEY}
     }
+
+    googleai-gemini {
+      model-name = "gemini-2.5-flash"
+      api-key = ${?GOOGLE_AI_GEMINI_API_KEY}
+    }
   }
 }
diff --git a/src/test/java/demo/multiagent/application/ActivityViewTest.java b/src/test/java/demo/multiagent/application/ActivityViewTest.java
index ebdf33a..bf58362 100644
--- a/src/test/java/demo/multiagent/application/ActivityViewTest.java
+++ b/src/test/java/demo/multiagent/application/ActivityViewTest.java
@@ -19,12 +19,12 @@ protected TestKit.Settings testKitSettings() {
     // We don't need a real one though.
     return TestKit.Settings.DEFAULT.withAdditionalConfig(
       ConfigFactory.parseString("akka.javasdk.agent.openai.api-key=fake-key")
-    ).withWorkflowIncomingMessages("agent-team");
+    ).withWorkflowIncomingMessages(AgentTeamWorkflow.class);
   }
 
   @Test
   public void shouldHandleMultipleSessionsForSameUser() {
-    var workflowMessages = testKit.getWorkflowIncomingMessages("agent-team");
+    var workflowMessages = testKit.getWorkflowIncomingMessages(AgentTeamWorkflow.class);
 
     var userId = "alice";
 
@@ -102,7 +102,7 @@ public void shouldReturnEmptyForNonExistentUser() {
 
   @Test
   public void shouldFilterActivitiesByUserId() {
-    var workflowMessages = testKit.getWorkflowIncomingMessages("agent-team");
+    var workflowMessages = testKit.getWorkflowIncomingMessages(AgentTeamWorkflow.class);
 
     // Activity for user bob
     var aliceState = new AgentTeamWorkflow.State(