diff --git a/.ci/containers/downstream-builder/generate_downstream.sh b/.ci/containers/downstream-builder/generate_downstream.sh
index 57abca7b7828..3c4eaf0c9bbe 100755
--- a/.ci/containers/downstream-builder/generate_downstream.sh
+++ b/.ci/containers/downstream-builder/generate_downstream.sh
@@ -28,7 +28,7 @@ function clone_repo() {
         GH_REPO=terraform-google-conversion
         LOCAL_PATH=$GOPATH/src/github.com/GoogleCloudPlatform/terraform-google-conversion
     elif [ "$REPO" == "tf-oics" ]; then
-        if [ "$UPSTREAM_BRANCH" == "main"]; then
+        if [ "$UPSTREAM_BRANCH" == "main" ]; then
             UPSTREAM_BRANCH=master
         fi
         UPSTREAM_OWNER=terraform-google-modules
@@ -48,7 +48,7 @@ function clone_repo() {
     mkdir -p "$(dirname $LOCAL_PATH)"
 
     echo "BASE_BRANCH: $BASE_BRANCH"
-    git clone $GITHUB_PATH $LOCAL_PATH --branch $BASE_BRANCH
+    git clone $GITHUB_PATH $LOCAL_PATH --branch $UPSTREAM_BRANCH
 }
 
 if [ $# -lt 4 ]; then
@@ -236,4 +236,4 @@ if [ "$COMMITTED" == "true" ] && [ "$COMMAND" == "downstream" ]; then
         "https://api.github.com/repos/$UPSTREAM_OWNER/$GH_REPO/pulls/$NEW_PR_NUMBER/merge"
 fi
 
-popd
+popd
\ No newline at end of file
diff --git a/.ci/containers/downstream-waiter/wait_for_commit.sh b/.ci/containers/downstream-waiter/wait_for_commit.sh
index f4a8e027422d..b59bde2a3698 100755
--- a/.ci/containers/downstream-waiter/wait_for_commit.sh
+++ b/.ci/containers/downstream-waiter/wait_for_commit.sh
@@ -10,7 +10,7 @@ SYNC_BRANCH_PREFIX=$1
 BASE_BRANCH=$2
 SHA=$3
 
-if [ "$BASE_BRANCH" == "main"]; then
+if [ "$BASE_BRANCH" == "main" ]; then
     SYNC_BRANCH=$SYNC_BRANCH_PREFIX
 else
     SYNC_BRANCH=$SYNC_BRANCH_PREFIX-$BASE_BRANCH
@@ -33,4 +33,4 @@ while true; do
         git fetch origin $SYNC_BRANCH
     fi
     sleep 5
-done
+done
\ No newline at end of file