diff --git a/scripts/util.sh b/scripts/util.sh
index ea35773..b4d1c84 100644
--- a/scripts/util.sh
+++ b/scripts/util.sh
@@ -96,6 +96,9 @@ clone_gitlab() (
     fi
 
     git clone $URL
+    cd $repo 
+    set_branch
+    git checkout $CODE_BRANCH 2>/dev/null || true
 )
 
 
@@ -111,6 +114,9 @@ clone_github() (
     fi
 
     git clone $URL
+    cd $repo
+    set_branch
+    git checkout $CODE_BRANCH 2>/dev/null || true
 )