diff --git a/infrastructure/terraform/lambda/fast-api/main.tf b/infrastructure/terraform/lambda/fast-api/main.tf
index fbe5805f..0cfd09f7 100644
--- a/infrastructure/terraform/lambda/fast-api/main.tf
+++ b/infrastructure/terraform/lambda/fast-api/main.tf
@@ -77,8 +77,8 @@ module "fastapi" {
 
   name        = "fastapi"
   stage       = var.stage
-  source_dir  = "${path.root}/../../../../backend"
-  handler     = "app.main.handler"
+  source_dir  = "${path.root}/../../../../"
+  handler     = "backend.app.main.handler"
   runtime     = "python3.11"
   timeout     = 600
   memory_size = 512
@@ -109,6 +109,8 @@ module "fastapi" {
     HEATING_KWH_PREDICTIONS_BUCKET  = data.terraform_remote_state.shared.outputs.retrofit_heating_kwh_predictions_bucket_name
     HOTWATER_KWH_PREDICTIONS_BUCKET = data.terraform_remote_state.shared.outputs.retrofit_hotwater_kwh_predictions_bucket_name
     ENERGY_ASSESSMENTS_BUCKET       = data.terraform_remote_state.shared.outputs.retrofit_energy_assessments_bucket_name
+    SECRET_KEY                      = data.terraform_remote_state.shared.outputs.secret_key
+    API_KEY                         = data.terraform_remote_state.shared.outputs.api_key
 
     ENGINE_SQS_URL         = data.terraform_remote_state.engine.outputs.ara_engine_queue_url
     CATEGORISATION_SQS_URL = data.terraform_remote_state.categorisation.outputs.categorisation_queue_url
@@ -120,7 +122,7 @@ module "fastapi" {
 ############################################
 resource "aws_iam_role_policy_attachment" "fast_api_s3_read" {
   role       = module.fastapi.role_name
-  policy_arn = data.terraform_remote_state.shared.outputs.fast_api_s3_read_and_write_arn
+  policy_arn = data.terraform_remote_state.shared.outputs.fast_api_s3_read_arn
 }
 
 module "fastapi_sqs_policy" {