diff --git a/infrastructure/ecs.tf b/infrastructure/ecs.tf
index f2935940..76512d28 100644
--- a/infrastructure/ecs.tf
+++ b/infrastructure/ecs.tf
@@ -10,7 +10,6 @@ module "ecs" {
   ecr_repository_uri = var.ecr_repository_uri
   ecs_cluster_id     = data.terraform_remote_state.platform.outputs.ecs_cluster_id
   health_check       = local.health_check
-  target_group_id    = aws_lb_target_group.target_group.id
   target_group_arn   = aws_lb_target_group.target_group.arn
   environment_variables = {
     "ENVIRONMENT"           = terraform.workspace,
diff --git a/infrastructure/load_balancer.tf b/infrastructure/load_balancer.tf
index e7a816c8..678dcb21 100644
--- a/infrastructure/load_balancer.tf
+++ b/infrastructure/load_balancer.tf
@@ -1,13 +1,11 @@
 module "load_balancer" {
-  source           = "../../i-ai-core-infrastructure/modules/load_balancer"
-  project          = var.project_name
-  region           = var.region
-  state_bucket     = var.state_bucket
-  account_id       = var.account_id
-  vpc_id           = data.terraform_remote_state.vpc.outputs.vpc_id
-  public_subnets   = data.terraform_remote_state.vpc.outputs.public_subnets
-  certificate_arn  = data.terraform_remote_state.universal.outputs.certificate_arn
-  ip_whitelist     = concat(var.internal_ips, var.developer_ips, var.external_ips)
-  env              = var.env
-  target_group_arn = aws_lb_target_group.target_group.arn
+  source         = "../../i-ai-core-infrastructure/modules/load_balancer"
+  project        = var.project_name
+  region         = var.region
+  state_bucket   = var.state_bucket
+  account_id     = var.account_id
+  vpc_id         = data.terraform_remote_state.vpc.outputs.vpc_id
+  public_subnets = data.terraform_remote_state.vpc.outputs.public_subnets
+  ip_whitelist   = concat(var.internal_ips, var.developer_ips, var.external_ips)
+  env            = var.env
 }