diff --git a/src/command_modules/azure-cli-batchai/HISTORY.rst b/src/command_modules/azure-cli-batchai/HISTORY.rst
index f74de6601a8..f3bff83a11a 100644
--- a/src/command_modules/azure-cli-batchai/HISTORY.rst
+++ b/src/command_modules/azure-cli-batchai/HISTORY.rst
@@ -3,6 +3,10 @@
 Release History
 ===============
 
+0.4.3
++++++
+* Fix auto storage account creation failure in `az batchai cluster create` command
+
 0.4.2
 +++++
 * Minor fixes
diff --git a/src/command_modules/azure-cli-batchai/azure/cli/command_modules/batchai/custom.py b/src/command_modules/azure-cli-batchai/azure/cli/command_modules/batchai/custom.py
index fcaeb6530a7..c70ded100da 100644
--- a/src/command_modules/azure-cli-batchai/azure/cli/command_modules/batchai/custom.py
+++ b/src/command_modules/azure-cli-batchai/azure/cli/command_modules/batchai/custom.py
@@ -575,7 +575,7 @@ def _create_auto_storage_account(storage_client, resource_group, location):
         name = _generate_auto_storage_account_name()
         check = storage_client.storage_accounts.check_name_availability(name).name_available
     storage_client.storage_accounts.create(resource_group, name, {
-        'sku': Sku(SkuName.standard_lrs),
+        'sku': Sku(name=SkuName.standard_lrs),
         'kind': Kind.storage,
         'location': location}).result()
     return name
diff --git a/src/command_modules/azure-cli-batchai/setup.py b/src/command_modules/azure-cli-batchai/setup.py
index b8d5d9da839..03d28cea2da 100644
--- a/src/command_modules/azure-cli-batchai/setup.py
+++ b/src/command_modules/azure-cli-batchai/setup.py
@@ -13,7 +13,7 @@
     logger.warn("Wheel is not available, disabling bdist_wheel hook")
     cmdclass = {}
 
-VERSION = "0.4.2"
+VERSION = "0.4.3"
 # The full list of classifiers is available at
 # https://pypi.python.org/pypi?%3Aaction=list_classifiers
 CLASSIFIERS = [