diff --git a/src/routes/(console)/project-[region]-[project]/functions/function-[function]/settings/updateRepository.svelte b/src/routes/(console)/project-[region]-[project]/functions/function-[function]/settings/updateRepository.svelte
index e4192ecee..650daf369 100644
--- a/src/routes/(console)/project-[region]-[project]/functions/function-[function]/settings/updateRepository.svelte
+++ b/src/routes/(console)/project-[region]-[project]/functions/function-[function]/settings/updateRepository.svelte
@@ -274,5 +274,9 @@
     <DisconnectRepo bind:show={showDisconnect} on:success={loadRepository} />
 {/if}
 {#if showSelectRoot}
-    <SelectRootModal bind:show={showSelectRoot} product="sites" bind:rootDir={selectedDir} />
+    <SelectRootModal
+        bind:show={showSelectRoot}
+        product="sites"
+        bind:rootDir={selectedDir}
+        branch={selectedBranch} />
 {/if}
diff --git a/src/routes/(console)/project-[region]-[project]/sites/site-[site]/settings/updateRepository.svelte b/src/routes/(console)/project-[region]-[project]/sites/site-[site]/settings/updateRepository.svelte
index 4d9579b2d..7c0c376da 100644
--- a/src/routes/(console)/project-[region]-[project]/sites/site-[site]/settings/updateRepository.svelte
+++ b/src/routes/(console)/project-[region]-[project]/sites/site-[site]/settings/updateRepository.svelte
@@ -269,5 +269,9 @@
 {/if}
 
 {#if showSelectRoot}
-    <SelectRootModal bind:show={showSelectRoot} product="sites" bind:rootDir={selectedDir} />
+    <SelectRootModal
+        bind:show={showSelectRoot}
+        product="sites"
+        bind:rootDir={selectedDir}
+        branch={selectedBranch} />
 {/if}