diff --git a/infrastructure/terraform/lambda/condition-etl/main.tf b/infrastructure/terraform/lambda/condition-etl/main.tf
index 98e97a08..0128f975 100644
--- a/infrastructure/terraform/lambda/condition-etl/main.tf
+++ b/infrastructure/terraform/lambda/condition-etl/main.tf
@@ -2,19 +2,20 @@ data "aws_secretsmanager_secret_version" "db_credentials" {
   secret_id = "${var.stage}/assessment_model/db_credentials"
 }
 
-locals {
-  db_credentials = jsondecode(data.aws_secretsmanager_secret_version.db_credentials.secret_string)
-}
-
 data "terraform_remote_state" "shared" {
   backend = "s3"
   config = {
     bucket = "assessment-model-terraform-state"
-    key = "env:/dev/terraform.tfstate" # TODO: dont hardcode this
+    key = "env:/${var.stage}/terraform.tfstate" # TODO: dont hardcode this
     region = "eu-west-2"
   }
 }
 
+locals {
+  db_credentials = jsondecode(data.aws_secretsmanager_secret_version.db_credentials.secret_string)
+}
+
+
 module "lambda" {
   source = "../modules/lambda_with_sqs"