diff --git a/src/test/java/com/google/genai/TuningsTest.java b/src/test/java/com/google/genai/TuningsTest.java
index 3f26067de60..f9ad4dbd2d4 100644
--- a/src/test/java/com/google/genai/TuningsTest.java
+++ b/src/test/java/com/google/genai/TuningsTest.java
@@ -254,6 +254,8 @@ public void testTuneWithEvaluationConfig(boolean vertexAI) {
             .autoraterConfig(
                 AutoraterConfig.builder().autoraterModel("test-model").samplingCount(1).build())
             .metrics(metrics)
+            .inferenceGenerationConfig(
+                GenerationConfig.builder().temperature(0.5f).maxOutputTokens(1024).build())
             .build();
     CreateTuningJobConfig tuningConfig =
         CreateTuningJobConfig.builder()
@@ -277,6 +279,7 @@ public void testTuneWithEvaluationConfig(boolean vertexAI) {
     // Assert
     assertNotNull(currentJob);
     assertTrue(currentJob.evaluationConfig().isPresent());
+    assertTrue(currentJob.evaluationConfig().get().inferenceGenerationConfig().isPresent());
     assertTrue(currentJob.state().get().knownEnum() == JobState.Known.JOB_STATE_PENDING);
   }
 }