diff --git a/main.tf b/main.tf
index 67c9bfb..8218b51 100644
--- a/main.tf
+++ b/main.tf
@@ -208,7 +208,7 @@ resource "azurerm_private_dns_zone" "flexible_postgres" {
   count = var.private_dns_zones.flexible_postgres == null ? 1 : 0
 
   name                = "${var.resource_prefix}.nebuly.postgres.database.azure.com"
-  resource_group_name = data.azurerm_resource_group.main
+  resource_group_name = data.azurerm_resource_group.main.name
 }
 resource "azurerm_private_dns_zone_virtual_network_link" "flexible_postgres" {
   count = var.private_dns_zones.flexible_postgres == null ? 1 : 0
@@ -225,7 +225,7 @@ resource "azurerm_private_dns_zone_virtual_network_link" "flexible_postgres" {
 resource "azurerm_private_dns_zone" "key_vault" {
   count               = var.private_dns_zones.key_vault == null ? 1 : 0
   name                = "privatelink.vaultcore.azure.net"
-  resource_group_name = var.private_dns_zones.key_vault.resource_group_name
+  resource_group_name = data.azurerm_resource_group.main.name
 }
 resource "azurerm_private_dns_zone_virtual_network_link" "key_vault" {
   count = var.private_dns_zones.key_vault == null ? 1 : 0