diff --git a/vpn-transit-gateway/main.tf b/vpn-transit-gateway/main.tf
index a5d2f54..96dcba4 100644
--- a/vpn-transit-gateway/main.tf
+++ b/vpn-transit-gateway/main.tf
@@ -122,7 +122,7 @@ resource "random_string" "tunnel_preshared_key" {
 #---
 resource "aws_vpn_connection" "vpn" {
   for_each = var.create ? local.vpn_settings : {}
-  type     = aws_customer_gateway.vpn[each.key]
+  type     = aws_customer_gateway.vpn[each.key].type
 
   transit_gateway_id  = var.transit_gateway_id
   customer_gateway_id = aws_customer_gateway.vpn[each.key].id