diff --git a/azure-mgmt-batchai/azure/mgmt/batchai/models/batch_ai_management_client_enums.py b/azure-mgmt-batchai/azure/mgmt/batchai/models/batch_ai_management_client_enums.py
index 5eb2b6a67bfe..e29a03a48fad 100644
--- a/azure-mgmt-batchai/azure/mgmt/batchai/models/batch_ai_management_client_enums.py
+++ b/azure-mgmt-batchai/azure/mgmt/batchai/models/batch_ai_management_client_enums.py
@@ -81,7 +81,7 @@ class ToolType(str, Enum):
     caffe2 = "caffe2"
     chainer = "chainer"
     horovod = "horovod"
-    mpi = "mpi"
+    custommpi = "custommpi"
     custom = "custom"
 
 
diff --git a/azure-mgmt-batchai/azure/mgmt/batchai/models/job.py b/azure-mgmt-batchai/azure/mgmt/batchai/models/job.py
index 8456615de021..cdec426d77d7 100644
--- a/azure-mgmt-batchai/azure/mgmt/batchai/models/job.py
+++ b/azure-mgmt-batchai/azure/mgmt/batchai/models/job.py
@@ -45,9 +45,9 @@ class Job(ProxyResource):
      on the VM.
     :type container_settings: ~azure.mgmt.batchai.models.ContainerSettings
     :param tool_type: The toolkit type of this job. Possible values are: cntk,
-     tensorflow, caffe, caffe2, chainer, pytorch, custom, mpi, horovod.
+     tensorflow, caffe, caffe2, chainer, pytorch, custom, custommpi, horovod.
      Possible values include: 'cntk', 'tensorflow', 'caffe', 'caffe2',
-     'chainer', 'horovod', 'mpi', 'custom'
+     'chainer', 'horovod', 'custommpi', 'custom'
     :type tool_type: str or ~azure.mgmt.batchai.models.ToolType
     :param cntk_settings: Specifies the settings for CNTK (aka Microsoft
      Cognitive Toolkit) job.
diff --git a/azure-mgmt-batchai/azure/mgmt/batchai/models/job_py3.py b/azure-mgmt-batchai/azure/mgmt/batchai/models/job_py3.py
index 9b1a5af01c6a..f44c7993df5d 100644
--- a/azure-mgmt-batchai/azure/mgmt/batchai/models/job_py3.py
+++ b/azure-mgmt-batchai/azure/mgmt/batchai/models/job_py3.py
@@ -45,9 +45,9 @@ class Job(ProxyResource):
      on the VM.
     :type container_settings: ~azure.mgmt.batchai.models.ContainerSettings
     :param tool_type: The toolkit type of this job. Possible values are: cntk,
-     tensorflow, caffe, caffe2, chainer, pytorch, custom, mpi, horovod.
+     tensorflow, caffe, caffe2, chainer, pytorch, custom, custommpi, horovod.
      Possible values include: 'cntk', 'tensorflow', 'caffe', 'caffe2',
-     'chainer', 'horovod', 'mpi', 'custom'
+     'chainer', 'horovod', 'custommpi', 'custom'
     :type tool_type: str or ~azure.mgmt.batchai.models.ToolType
     :param cntk_settings: Specifies the settings for CNTK (aka Microsoft
      Cognitive Toolkit) job.