diff --git a/src/azure-cli/azure/cli/command_modules/vm/aaz/latest/vm/__init__.py b/src/azure-cli/azure/cli/command_modules/vm/aaz/latest/vm/__init__.py
index bd8a551b579..33f6342d7a8 100644
--- a/src/azure-cli/azure/cli/command_modules/vm/aaz/latest/vm/__init__.py
+++ b/src/azure-cli/azure/cli/command_modules/vm/aaz/latest/vm/__init__.py
@@ -28,9 +28,9 @@
 from ._reapply import *
 from ._redeploy import *
 from ._reimage import *
+from ._restart import *
 from ._show import *
 from ._simulate_eviction import *
 from ._start import *
 from ._update import *
 from ._wait import *
-from ._restart import *
\ No newline at end of file
diff --git a/src/azure-cli/azure/cli/command_modules/vm/aaz/latest/vmss/__init__.py b/src/azure-cli/azure/cli/command_modules/vm/aaz/latest/vmss/__init__.py
index a1c7051c25b..f7d86e4ec18 100644
--- a/src/azure-cli/azure/cli/command_modules/vm/aaz/latest/vmss/__init__.py
+++ b/src/azure-cli/azure/cli/command_modules/vm/aaz/latest/vmss/__init__.py
@@ -19,10 +19,11 @@
 from ._list_instances import *
 from ._list_skus import *
 from ._patch import *
+from ._manualupgrade import *
 from ._perform_maintenance import *
-from ._restart import *
 from ._reimage import *
 from ._reimageall import *
+from ._restart import *
 from ._set_orchestration_service_state import *
 from ._show import *
 from ._simulate_eviction import *
@@ -31,4 +32,3 @@
 from ._update import *
 from ._update_domain_walk import *
 from ._wait import *
-from ._manualupgrade import *