diff --git a/terraform/api.tf b/terraform/api.tf
index 2cb8b8b..bde99bc 100644
--- a/terraform/api.tf
+++ b/terraform/api.tf
@@ -699,7 +699,7 @@ resource "aws_codepipeline" "leafsteroids_api" {
     action {
       category = "Source"
       configuration = {
-        "BranchName"           = "feature/dockerize"
+        "BranchName"           = var.branch_name
         "ConnectionArn"        = var.connection_arn
         "DetectChanges"        = "false"
         "FullRepositoryId"     = "mongodb-developer/leafsteroids"
diff --git a/terraform/variables.tf b/terraform/variables.tf
index 40fe155..ffae909 100644
--- a/terraform/variables.tf
+++ b/terraform/variables.tf
@@ -63,3 +63,8 @@ variable "game_client_port" {
   description = "The port of the game client"
   default     = 80
 }
+
+variable "branch_name" {
+  description = "The name of the branch"
+}
+
diff --git a/terraform/website.tf b/terraform/website.tf
index a77a771..d06f4d0 100644
--- a/terraform/website.tf
+++ b/terraform/website.tf
@@ -731,7 +731,7 @@ resource "aws_codepipeline" "leafsteroids_website" {
     action {
       category = "Source"
       configuration = {
-        "BranchName"           = "feature/dockerize"
+        "BranchName"           = var.branch_name
         "ConnectionArn"        = var.connection_arn
         "DetectChanges"        = "false"
         "FullRepositoryId"     = "mongodb-developer/leafsteroids"