diff --git a/apps/desktop/src/lib/components/BaseBranch.svelte b/apps/desktop/src/lib/components/BaseBranch.svelte
index acc82d2e78..23f8b3d729 100644
--- a/apps/desktop/src/lib/components/BaseBranch.svelte
+++ b/apps/desktop/src/lib/components/BaseBranch.svelte
@@ -1,10 +1,10 @@
 <script lang="ts">
+	import IntegrateUpstreamModal from './IntegrateUpstreamModal.svelte';
 	import { Project } from '$lib/backend/projects';
 	import { BaseBranchService } from '$lib/baseBranch/baseBranchService';
 	import CommitAction from '$lib/commit/CommitAction.svelte';
 	import CommitCard from '$lib/commit/CommitCard.svelte';
 	import { transformAnyCommit } from '$lib/commitLines/transformers';
-	import UpdateBaseButton from '$lib/components/UpdateBaseButton.svelte';
 	import { projectMergeUpstreamWarningDismissed } from '$lib/config/config';
 	import { getGitHost } from '$lib/gitHost/interface/gitHost';
 	import { ModeService } from '$lib/modes/service';
@@ -67,9 +67,9 @@
 	let baseBranchIsUpdating = $state<boolean>(false);
 	let updateTargetModal = $state<Modal>();
 	let confirmResetModal = $state<Modal>();
+	let integrateUpstreamModal = $state<IntegrateUpstreamModal | undefined>();
 	let resetActionType = $state<ResetActionType | undefined>(undefined);
 	let mergeUpstreamWarningDismissedCheckbox = $state<boolean>(false);
-	let updateBaseButton = $state<UpdateBaseButton | undefined>();
 
 	const multiple = $derived(
 		base ? base.upstreamCommits.length > 1 || base.upstreamCommits.length === 0 : false
@@ -196,6 +196,8 @@
 			{multiple ? 'commits' : 'commit'}
 		</div>
 
+		<IntegrateUpstreamModal bind:this={integrateUpstreamModal} />
+
 		<Button
 			style="pop"
 			kind="solid"
@@ -203,7 +205,7 @@
 			disabled={$mode?.type !== 'OpenWorkspace'}
 			onclick={() => {
 				if (project.succeedingRebases) {
-					updateBaseButton?.openModal();
+					integrateUpstreamModal?.show();
 				} else {
 					if ($mergeUpstreamWarningDismissed) {
 						updateBaseBranch();