diff --git a/src/azure-cli/azure/cli/command_modules/apim/_params.py b/src/azure-cli/azure/cli/command_modules/apim/_params.py
index 65bd6ce6260..4aa8eec5eab 100644
--- a/src/azure-cli/azure/cli/command_modules/apim/_params.py
+++ b/src/azure-cli/azure/cli/command_modules/apim/_params.py
@@ -8,7 +8,7 @@
                                                 get_location_type,
                                                 resource_group_name_type,
                                                 get_three_state_flag)
-from azure.mgmt.apimanagement.models.api_management_client_enums import (SkuType, VirtualNetworkType)
+from azure.mgmt.apimanagement.models import (SkuType, VirtualNetworkType)
 
 
 SKU_TYPES = SkuType