diff --git a/infrastructure/terraform/lambda/_template/variables.tf b/infrastructure/terraform/lambda/_template/variables.tf
index 7ba0bedb..5876cf3a 100644
--- a/infrastructure/terraform/lambda/_template/variables.tf
+++ b/infrastructure/terraform/lambda/_template/variables.tf
@@ -7,3 +7,8 @@ variable "image_uri" {
   type        = string
   description = "Full ECR image URI including digest"
 }
+
+variable "stage" {
+  description = "Deployment stage (e.g. dev, prod)"
+  type        = string
+}
\ No newline at end of file
diff --git a/infrastructure/terraform/lambda/address2UPRN/variables.tf b/infrastructure/terraform/lambda/address2UPRN/variables.tf
index 7ba0bedb..5876cf3a 100644
--- a/infrastructure/terraform/lambda/address2UPRN/variables.tf
+++ b/infrastructure/terraform/lambda/address2UPRN/variables.tf
@@ -7,3 +7,8 @@ variable "image_uri" {
   type        = string
   description = "Full ECR image URI including digest"
 }
+
+variable "stage" {
+  description = "Deployment stage (e.g. dev, prod)"
+  type        = string
+}
\ No newline at end of file