diff --git a/infrastructure/terraform/lambda/fast-api/main.tf b/infrastructure/terraform/lambda/fast-api/main.tf
index 82d2cc60..ebf436c3 100644
--- a/infrastructure/terraform/lambda/fast-api/main.tf
+++ b/infrastructure/terraform/lambda/fast-api/main.tf
@@ -1,3 +1,6 @@
+############################################
+# Load Terraform State
+############################################
 data "terraform_remote_state" "shared" {
   backend = "s3"
   config = {
@@ -25,6 +28,9 @@ data "terraform_remote_state" "categorisation" {
   }
 }
 
+############################################
+# Load Credentials
+############################################
 data "aws_secretsmanager_secret_version" "db_credentials" {
   secret_id = "${var.stage}/assessment_model/db_credentials"
 }