diff --git a/sdk/datamigration/azure-mgmt-datamigration/azure/mgmt/datamigration/models/_models.py b/sdk/datamigration/azure-mgmt-datamigration/azure/mgmt/datamigration/models/_models.py
index cf3a9c11d52d..2d2e1d09c92d 100644
--- a/sdk/datamigration/azure-mgmt-datamigration/azure/mgmt/datamigration/models/_models.py
+++ b/sdk/datamigration/azure-mgmt-datamigration/azure/mgmt/datamigration/models/_models.py
@@ -288,28 +288,27 @@ class CheckOCIDriverTaskOutput(Model):
     Variables are only populated by the server, and will be ignored when
     sending a request.
 
-    :ivar installed_driver: Information about the installed driver if found
+    :param installed_driver: Information about the installed driver if found
      and valid.
-    :vartype installed_driver: dict[str,
-     list[~azure.mgmt.datamigration.models.OracleOCIDriverInfo]]
+    :type installed_driver:
+     ~azure.mgmt.datamigration.models.OracleOCIDriverInfo
     :ivar validation_errors: Validation errors
     :vartype validation_errors:
      list[~azure.mgmt.datamigration.models.ReportableException]
     """
 
     _validation = {
-        'installed_driver': {'readonly': True},
         'validation_errors': {'readonly': True},
     }
 
     _attribute_map = {
-        'installed_driver': {'key': 'installedDriver', 'type': '{[OracleOCIDriverInfo]}'},
+        'installed_driver': {'key': 'installedDriver', 'type': 'OracleOCIDriverInfo'},
         'validation_errors': {'key': 'validationErrors', 'type': '[ReportableException]'},
     }
 
     def __init__(self, **kwargs):
         super(CheckOCIDriverTaskOutput, self).__init__(**kwargs)
-        self.installed_driver = None
+        self.installed_driver = kwargs.get('installed_driver', None)
         self.validation_errors = None
 
 
diff --git a/sdk/datamigration/azure-mgmt-datamigration/azure/mgmt/datamigration/models/_models_py3.py b/sdk/datamigration/azure-mgmt-datamigration/azure/mgmt/datamigration/models/_models_py3.py
index 5a546a581e97..6c484a582310 100644
--- a/sdk/datamigration/azure-mgmt-datamigration/azure/mgmt/datamigration/models/_models_py3.py
+++ b/sdk/datamigration/azure-mgmt-datamigration/azure/mgmt/datamigration/models/_models_py3.py
@@ -288,28 +288,27 @@ class CheckOCIDriverTaskOutput(Model):
     Variables are only populated by the server, and will be ignored when
     sending a request.
 
-    :ivar installed_driver: Information about the installed driver if found
+    :param installed_driver: Information about the installed driver if found
      and valid.
-    :vartype installed_driver: dict[str,
-     list[~azure.mgmt.datamigration.models.OracleOCIDriverInfo]]
+    :type installed_driver:
+     ~azure.mgmt.datamigration.models.OracleOCIDriverInfo
     :ivar validation_errors: Validation errors
     :vartype validation_errors:
      list[~azure.mgmt.datamigration.models.ReportableException]
     """
 
     _validation = {
-        'installed_driver': {'readonly': True},
         'validation_errors': {'readonly': True},
     }
 
     _attribute_map = {
-        'installed_driver': {'key': 'installedDriver', 'type': '{[OracleOCIDriverInfo]}'},
+        'installed_driver': {'key': 'installedDriver', 'type': 'OracleOCIDriverInfo'},
         'validation_errors': {'key': 'validationErrors', 'type': '[ReportableException]'},
     }
 
-    def __init__(self, **kwargs) -> None:
+    def __init__(self, *, installed_driver=None, **kwargs) -> None:
         super(CheckOCIDriverTaskOutput, self).__init__(**kwargs)
-        self.installed_driver = None
+        self.installed_driver = installed_driver
         self.validation_errors = None