diff --git a/infrastructure/terraform/lambda/address2UPRN/provider.tf b/infrastructure/terraform/lambda/address2UPRN/provider.tf
index 2f4360ec..36cb63f1 100644
--- a/infrastructure/terraform/lambda/address2UPRN/provider.tf
+++ b/infrastructure/terraform/lambda/address2UPRN/provider.tf
@@ -7,7 +7,7 @@ terraform {
   }
 
   backend "s3" {
-    bucket = "address2uprn-terraform-state"
+    bucket = var.state_bucket_name
     key    = "terraform.tfstate"
     region = "eu-west-2"
   }
diff --git a/infrastructure/terraform/lambda/address2UPRN/variables.tf b/infrastructure/terraform/lambda/address2UPRN/variables.tf
index 1d6c5952..b6f9907a 100644
--- a/infrastructure/terraform/lambda/address2UPRN/variables.tf
+++ b/infrastructure/terraform/lambda/address2UPRN/variables.tf
@@ -8,6 +8,10 @@ variable "stage" {
 }
 
 variable "image_digest" {
-  type        = string
-  description = "sha256 image digest from CI"
+  type = string
 }
+
+variable "state_bucket_name" {
+  type        = string
+  description = "S3 bucket name used for this lambda's Terraform state"
+}
\ No newline at end of file