diff --git a/main.tf b/main.tf
index 2eb31f7..e5e8ca6 100644
--- a/main.tf
+++ b/main.tf
@@ -16,7 +16,7 @@ resource "aws_vpc_endpoint" "s3" {
 }
 
 resource "aws_vpc_endpoint_route_table_association" "private_s3" {
-  count = var.enable_s3_endpoint ? local.nat_gateway_count : 0
+  count = var.enable_s3_endpoint ? 1 : 0
 
   vpc_endpoint_id = aws_vpc_endpoint.s3[0].id
   #route_table_id  = element(aws_route_table.private.*.id, count.index)