@@ -475,7 +475,7 @@ def _add_owner_reference(
         uid=owner_job.metadata.uid,
         controller=True,
     )
-    patch_body = {"metadata": {"ownerReferences": [owner_reference.to_dict()]}}
+    patch_body = {"metadata": {"ownerReferences": [owner_reference]}}
 
     try:
       self._core_v1.patch_namespaced_config_map(
 
@@ -52,6 +52,11 @@ def mock_k8s_clients():
   ) as mock_client_class:
     mock_batch_v1 = MagicMock(spec=client.BatchV1Api)
     mock_core_v1 = MagicMock(spec=client.CoreV1Api)
+    mock_batch_v1.create_namespaced_job.return_value = client.V1Job(
+        api_version="batch/v1",
+        kind="Job",
+        metadata=client.V1ObjectMeta(name="test-job", uid="test-job-uid"),
+    )
     mock_client_class.BatchV1Api.return_value = mock_batch_v1
     mock_client_class.CoreV1Api.return_value = mock_core_v1
     yield {
@@ -196,6 +201,33 @@ def test_execute_code_success(
     ].create_namespaced_config_map.assert_called_once()
     mock_k8s_clients["batch_v1"].create_namespaced_job.assert_called_once()
     mock_k8s_clients["core_v1"].patch_namespaced_config_map.assert_called_once()
+    body = mock_k8s_clients[
+        "core_v1"
+    ].patch_namespaced_config_map.call_args.kwargs["body"]
+    assert body == {
+        "metadata": {
+            "ownerReferences": [
+                client.V1OwnerReference(
+                    api_version="batch/v1",
+                    kind="Job",
+                    name="test-job",
+                    uid="test-job-uid",
+                    controller=True,
+                )
+            ]
+        }
+    }
+    assert client.ApiClient().sanitize_for_serialization(body) == {
+        "metadata": {
+            "ownerReferences": [{
+                "apiVersion": "batch/v1",
+                "controller": True,
+                "kind": "Job",
+                "name": "test-job",
+                "uid": "test-job-uid",
+            }]
+        }
+    }
     mock_k8s_clients["core_v1"].read_namespaced_pod_log.assert_called_once()
 
   @patch("google.adk.code_executors.gke_code_executor.Watch")
Original file line number	Diff line number	Diff line change
`@@ -475,7 +475,7 @@ def _add_owner_reference(`
`475`	`475`	`uid=owner_job.metadata.uid,`
`476`	`476`	`controller=True,`
`477`	`477`	`)`
`478`		`- patch_body = {"metadata": {"ownerReferences": [owner_reference.to_dict()]}}`
	`478`	`+ patch_body = {"metadata": {"ownerReferences": [owner_reference]}}`
`479`	`479`
`480`	`480`	`try:`
`481`	`481`	`self._core_v1.patch_namespaced_config_map(`