diff --git a/main/src/services/__tests__/gitStatusManager.test.ts b/main/src/services/__tests__/gitStatusManager.test.ts
index aae7cfc..b71c85c 100644
--- a/main/src/services/__tests__/gitStatusManager.test.ts
+++ b/main/src/services/__tests__/gitStatusManager.test.ts
@@ -85,6 +85,7 @@ describe('GitStatusManager', () => {
 
     mockWorktreeManager = {
       getProjectMainBranch: vi.fn().mockResolvedValue('main'),
+      getSessionComparisonBranch: vi.fn().mockResolvedValue('main'),
     } as Partial<WorktreeManager> as WorktreeManager;
 
     mockGitDiffManager = {} as Partial<GitDiffManager> as GitDiffManager;