diff --git a/examples/complete/data.tf b/examples/complete/data.tf
index 92c5376..4637d0f 100644
--- a/examples/complete/data.tf
+++ b/examples/complete/data.tf
@@ -1,6 +1,6 @@
 data "azurerm_client_config" "current" {}
 
 data "azurerm_api_management" "apim" {
-  name                = module.apim.apim_name
+  name                = module.apim.api_management.name
   resource_group_name = azurerm_resource_group.test_group.name
 }
diff --git a/examples/complete/main.tf b/examples/complete/main.tf
index 6064174..5a58ed9 100644
--- a/examples/complete/main.tf
+++ b/examples/complete/main.tf
@@ -93,8 +93,8 @@ resource "azurerm_key_vault_secret" "test_group" {
 module "apim" {
   source              = "../../"
   resource_group_name = azurerm_resource_group.test_group.name
-  prefix              = local.prefix
-  suffix              = local.suffix
+  prefix              = [local.unique_name_stub]
+  suffix              = [local.unique_name_stub]
 
   # API Management
   apim_publisher_name  = "John Doe"
@@ -130,8 +130,8 @@ module "apim" {
 resource "azurerm_key_vault_access_policy" "apim" {
   key_vault_id = azurerm_key_vault.test_group.id
 
-  tenant_id = module.apim.apim_identity[0].tenant_id
-  object_id = module.apim.apim_identity[0].principal_id
+  tenant_id = module.apim.api_management.identity[0].tenant_id
+  object_id = module.apim.api_management.identity[0].principal_id
 
   secret_permissions = [
     "get",