diff --git a/src/azure-cli/azure/cli/command_modules/ams/_params.py b/src/azure-cli/azure/cli/command_modules/ams/_params.py
index 404eed5f01e..c78242dc954 100644
--- a/src/azure-cli/azure/cli/command_modules/ams/_params.py
+++ b/src/azure-cli/azure/cli/command_modules/ams/_params.py
@@ -66,7 +66,7 @@ def load_arguments(self, _):  # pylint: disable=too-many-locals, too-many-statem
     with self.argument_context('ams account create') as c:
         c.argument('storage_account', storage_account_arg_type,
                    help='The name or resource ID of the primary storage account to attach to the Azure Media Services account. The storage account MUST be in the same Azure subscription as the Media Services account. It is strongly recommended that the storage account be in the same resource group as the Media Services account. Blob only accounts are not allowed as primary.')
-        c.argument('assign_identity', options_list=['--id-system-assigned'], action='store_true', help='Set the system managed identity on the media services account.')
+        c.argument('assign_identity', options_list=['--mi-system-assigned'], action='store_true', help='Set the system managed identity on the media services account.')
 
     with self.argument_context('ams account check-name') as c:
         c.argument('account_name', options_list=['--name', '-n'], id_part=None,
diff --git a/src/azure-cli/azure/cli/command_modules/ams/tests/latest/recordings/test_ams_create_show.yaml b/src/azure-cli/azure/cli/command_modules/ams/tests/latest/recordings/test_ams_create_show.yaml
index ef4c7735370..d5eddfaeeaf 100644
--- a/src/azure-cli/azure/cli/command_modules/ams/tests/latest/recordings/test_ams_create_show.yaml
+++ b/src/azure-cli/azure/cli/command_modules/ams/tests/latest/recordings/test_ams_create_show.yaml
@@ -16,7 +16,7 @@ interactions:
       Content-Type:
       - application/json; charset=utf-8
       ParameterSetName:
-      - -n -g --storage-account -l --id-system-assigned
+      - -n -g --storage-account -l --mi-system-assigned
       User-Agent:
       - python/3.8.2 (Windows-10-10.0.19041-SP0) msrest/0.6.18 msrest_azure/0.6.3
         azure-mgmt-media/3.0.0 Azure-SDK-For-Python AZURECLI/2.18.0
diff --git a/src/azure-cli/azure/cli/command_modules/ams/tests/latest/recordings/test_ams_encryption_set_show.yaml b/src/azure-cli/azure/cli/command_modules/ams/tests/latest/recordings/test_ams_encryption_set_show.yaml
index 5d8e9789733..e6b65ae15d7 100644
--- a/src/azure-cli/azure/cli/command_modules/ams/tests/latest/recordings/test_ams_encryption_set_show.yaml
+++ b/src/azure-cli/azure/cli/command_modules/ams/tests/latest/recordings/test_ams_encryption_set_show.yaml
@@ -16,7 +16,7 @@ interactions:
       Content-Type:
       - application/json; charset=utf-8
       ParameterSetName:
-      - -n -g --storage-account -l --id-system-assigned
+      - -n -g --storage-account -l --mi-system-assigned
       User-Agent:
       - python/3.8.2 (Windows-10-10.0.19041-SP0) msrest/0.6.18 msrest_azure/0.6.3
         azure-mgmt-media/3.0.0 Azure-SDK-For-Python AZURECLI/2.18.0
diff --git a/src/azure-cli/azure/cli/command_modules/ams/tests/latest/test_ams_account_encryption_scenarios.py b/src/azure-cli/azure/cli/command_modules/ams/tests/latest/test_ams_account_encryption_scenarios.py
index c639202e044..2249c5f4dd9 100644
--- a/src/azure-cli/azure/cli/command_modules/ams/tests/latest/test_ams_account_encryption_scenarios.py
+++ b/src/azure-cli/azure/cli/command_modules/ams/tests/latest/test_ams_account_encryption_scenarios.py
@@ -23,7 +23,7 @@ def test_ams_encryption_set_show(self, resource_group, storage_account_for_creat
             'identityPermissions': "get unwrapkey wrapkey",
         })
 
-        account_result = self.cmd('az ams account create -n {amsname} -g {rg} --storage-account {storageAccount} -l {location} --id-system-assigned', checks=[
+        account_result = self.cmd('az ams account create -n {amsname} -g {rg} --storage-account {storageAccount} -l {location} --mi-system-assigned', checks=[
             self.check('name', '{amsname}'),
             self.check('location', 'Central US')
         ])
diff --git a/src/azure-cli/azure/cli/command_modules/ams/tests/latest/test_ams_account_scenarios.py b/src/azure-cli/azure/cli/command_modules/ams/tests/latest/test_ams_account_scenarios.py
index 32ed8805e4a..0a9142faf32 100644
--- a/src/azure-cli/azure/cli/command_modules/ams/tests/latest/test_ams_account_scenarios.py
+++ b/src/azure-cli/azure/cli/command_modules/ams/tests/latest/test_ams_account_scenarios.py
@@ -18,7 +18,7 @@ def test_ams_create_show(self, resource_group, storage_account_for_create):
             'location': 'centralus',
         })
 
-        self.cmd('az ams account create -n {amsname} -g {rg} --storage-account {storageAccount} -l {location} --id-system-assigned', checks=[
+        self.cmd('az ams account create -n {amsname} -g {rg} --storage-account {storageAccount} -l {location} --mi-system-assigned', checks=[
             self.check('name', '{amsname}'),
             self.check('location', 'Central US'),
             self.check('identity.type', 'SystemAssigned')