diff --git a/infrastructure/terraform/lambda/fast-api/main.tf b/infrastructure/terraform/lambda/fast-api/main.tf
index f1a9494b..8dcbb8a3 100644
--- a/infrastructure/terraform/lambda/fast-api/main.tf
+++ b/infrastructure/terraform/lambda/fast-api/main.tf
@@ -31,8 +31,8 @@ data "terraform_remote_state" "categorisation" {
 data "terraform_remote_state" "postcode_splitter" {
   backend = "s3"
   config = {
-    bucket = "postcode-splitter-terraform-state"
-    key    = "terraform.tfstate"
+    bucket = "postcode-splitter-terraform-state",
+    key    = "env:/${var.stage}/terraform.tfstate"
     region = "eu-west-2"
   }
 }