diff --git a/modules/aws_ecs/locals.tf b/modules/aws_ecs/locals.tf
index 858a463..aa86cb7 100644
--- a/modules/aws_ecs/locals.tf
+++ b/modules/aws_ecs/locals.tf
@@ -8,6 +8,10 @@ locals {
       name  = "IS_ONPREM",
       value = "true"
     },
+    {
+      name  = "DEPLOYMENT_TEMPLATE_TYPE"
+      value = var.launch_type == "FARGATE" ? "aws-ecs-fargate-terraform" : "aws-ecs-ec2-terraform"
+    }
   ]
 
   // Use var.ecs_code_executor_image if defined, otherwise fallback to the same tag as var.ecs_retool_image
diff --git a/modules/aws_ecs_ec2/locals.tf b/modules/aws_ecs_ec2/locals.tf
index b45ed69..7a14150 100644
--- a/modules/aws_ecs_ec2/locals.tf
+++ b/modules/aws_ecs_ec2/locals.tf
@@ -45,6 +45,10 @@ locals {
       {
         "name" : "LICENSE_KEY",
         "value" : var.retool_license_key
+      },
+      {
+        "name": "DEPLOYMENT_TEMPLATE_TYPE"
+        "value": "aws-ecs-ec2-terraform-deprecated"
       }
     ]
   )