diff --git a/modules/container_app/dns.tf b/modules/container_app/dns.tf
index a79fd2d..71d2651 100644
--- a/modules/container_app/dns.tf
+++ b/modules/container_app/dns.tf
@@ -1,16 +1,18 @@
 data "azurerm_dns_zone" "dns_zone" {
+  count = var.custom_domain_count
   name = var.domain_name
 }
 
 data "azurerm_resource_group" "dns_zone_resource_group" {
+  count = var.custom_domain_count
   name = var.dns_zone_resource_group
 }
 
 resource "azurerm_dns_txt_record" "dns_txt_record" {
   count = var.custom_domain_count
   name = "asuid.${var.app_subdomain_name}"
-  resource_group_name = data.azurerm_resource_group.dns_zone_resource_group.name
-  zone_name = data.azurerm_dns_zone.dns_zone.name
+  resource_group_name = data.azurerm_resource_group.dns_zone_resource_group[0].name
+  zone_name = data.azurerm_dns_zone.dns_zone[0].name
   ttl = 14400
 
   record {