diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/_ml_client.py b/sdk/ml/azure-ai-ml/azure/ai/ml/_ml_client.py
index 345d4661cb29..0e3b3c7242ad 100644
--- a/sdk/ml/azure-ai-ml/azure/ai/ml/_ml_client.py
+++ b/sdk/ml/azure-ai-ml/azure/ai/ml/_ml_client.py
@@ -521,7 +521,7 @@ def __init__(
         self._batch_endpoints = BatchEndpointOperations(
             self._operation_scope,
             self._operation_config,
-            self._service_client_05_2022,
+            self._service_client_10_2023,
             self._operation_container,
             self._credential,
             requests_pipeline=self._requests_pipeline,
diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/_schema/_endpoint/batch/batch_endpoint_defaults.py b/sdk/ml/azure-ai-ml/azure/ai/ml/_schema/_endpoint/batch/batch_endpoint_defaults.py
index db4c986eb0a1..49699bb04416 100644
--- a/sdk/ml/azure-ai-ml/azure/ai/ml/_schema/_endpoint/batch/batch_endpoint_defaults.py
+++ b/sdk/ml/azure-ai-ml/azure/ai/ml/_schema/_endpoint/batch/batch_endpoint_defaults.py
@@ -9,7 +9,7 @@
 
 from marshmallow import fields, post_load
 
-from azure.ai.ml._restclient.v2022_05_01.models import BatchEndpointDefaults
+from azure.ai.ml._restclient.v2023_10_01.models import BatchEndpointDefaults
 from azure.ai.ml._schema.core.schema import PatchedSchemaMeta
 
 module_logger = logging.getLogger(__name__)