diff --git a/vpc-transit-gateway-association/README.md b/vpc-transit-gateway-association/README.md
index b87388e..6a39251 100644
--- a/vpc-transit-gateway-association/README.md
+++ b/vpc-transit-gateway-association/README.md
@@ -72,7 +72,7 @@ This sets up the necessary transit gateway configuration for attaching a VPC wit
 | <a name="input_account_id"></a> [account\_id](#input\_account\_id) | AWS Account ID (default: will pull from current user) | `string` | `""` | no |
 | <a name="input_availability_zones"></a> [availability\_zones](#input\_availability\_zones) | AWS Availability Zones to use (by default will use all available) | `list(string)` | `[]` | no |
 | <a name="input_create"></a> [create](#input\_create) | Flag to indicate whether to create the resources or not (default: true) | `bool` | `true` | no |
-| <a name="input_data_input"></a> [data\_input](#input\_data\_input) | Map of data generated by vpc-transit-gateway-association-data | <pre>object({<br>    availablity_zone      = map(string)<br>    gateway_self          = string<br>    gateway_peer          = string<br>    route_tables_self     = map(object())<br>    route_tables_peer     = map(object())<br>    map_route_tables_self = map(map(string))<br>    map_route_tables_peer = map(map(string))<br>    prefix_list_id_ipv4   = string<br>  })</pre> | n/a | yes |
+| <a name="input_data_input"></a> [data\_input](#input\_data\_input) | Map of data generated by vpc-transit-gateway-association-data | <pre>object({<br>    availablity_zone      = map(string)<br>    gateway_self          = string<br>    gateway_peer          = string<br>    route_tables_self     = map(map(string))<br>    route_tables_peer     = map(map(string))<br>    map_route_tables_self = map(map(string))<br>    map_route_tables_peer = map(map(string))<br>    prefix_list_id_ipv4   = string<br>  })</pre> | n/a | yes |
 | <a name="input_network_account_profile"></a> [network\_account\_profile](#input\_network\_account\_profile) | AWS profile of the source account sharing the VPC resources | `string` | n/a | yes |
 | <a name="input_override_prefixes"></a> [override\_prefixes](#input\_override\_prefixes) | Override built-in prefixes by component. This should be used primarily for common infrastructure things | `map(string)` | `{}` | no |
 | <a name="input_private_subnets_ids"></a> [private\_subnets\_ids](#input\_private\_subnets\_ids) | List of private subnet objects including: subnet, label, availability\_zone, id | <pre>list(object({<br>    subnet            = string<br>    label             = string<br>    availability_zone = string<br>    id                = string<br>  }))</pre> | `[]` | no |
diff --git a/vpc-transit-gateway-association/variables.data_input.tf b/vpc-transit-gateway-association/variables.data_input.tf
index 425b9e3..d0e2589 100644
--- a/vpc-transit-gateway-association/variables.data_input.tf
+++ b/vpc-transit-gateway-association/variables.data_input.tf
@@ -4,8 +4,8 @@ variable "data_input" {
     availablity_zone      = map(string)
     gateway_self          = string
     gateway_peer          = string
-    route_tables_self     = map(object())
-    route_tables_peer     = map(object())
+    route_tables_self     = map(map(string))
+    route_tables_peer     = map(map(string))
     map_route_tables_self = map(map(string))
     map_route_tables_peer = map(map(string))
     prefix_list_id_ipv4   = string