diff --git a/common/src/test/java/com/google/ai/client/generativeai/common/UnarySnapshotTests.kt b/common/src/test/java/com/google/ai/client/generativeai/common/UnarySnapshotTests.kt
index b4da178e..ad5c634a 100644
--- a/common/src/test/java/com/google/ai/client/generativeai/common/UnarySnapshotTests.kt
+++ b/common/src/test/java/com/google/ai/client/generativeai/common/UnarySnapshotTests.kt
@@ -348,7 +348,7 @@ internal class UnarySnapshotTests {
         val codeExecutionResult = content.parts[1]
 
         executableCodePart.shouldBe(
-          ExecutableCodePart(ExecutableCode("python", "print(\"Hello World\")"))
+          ExecutableCodePart(ExecutableCode("PYTHON", "print(\"Hello World\")"))
         )
         codeExecutionResult.shouldBe(
           CodeExecutionResultPart(CodeExecutionResult(Outcome.OUTCOME_OK, "Hello World"))
diff --git a/common/src/test/resources/golden-files/unary/success-code-execution.json b/common/src/test/resources/golden-files/unary/success-code-execution.json
index 1f40e022..3b8f4c25 100644
--- a/common/src/test/resources/golden-files/unary/success-code-execution.json
+++ b/common/src/test/resources/golden-files/unary/success-code-execution.json
@@ -5,7 +5,7 @@
         "parts": [
           {
             "executableCode": {
-              "language": "python",
+              "language": "PYTHON",
               "code": "print(\"Hello World\")"
             }
           },