diff --git a/src/main/github/client-pr-local-runtime.test.ts b/src/main/github/client-pr-local-runtime.test.ts
index 543d6f239f2..0633b137b1c 100644
--- a/src/main/github/client-pr-local-runtime.test.ts
+++ b/src/main/github/client-pr-local-runtime.test.ts
@@ -85,6 +85,7 @@ vi.mock('./github-enterprise-repository', () => ({
 }))
 
 vi.mock('../providers/ssh-git-dispatch', () => ({
+  getSshGitProviderGeneration: vi.fn(() => 0),
   getSshGitProvider: vi.fn()
 }))
 
diff --git a/src/main/github/client-test-mocks.ts b/src/main/github/client-test-mocks.ts
index c27af20defe..15fe0f36214 100644
--- a/src/main/github/client-test-mocks.ts
+++ b/src/main/github/client-test-mocks.ts
@@ -132,12 +132,14 @@ export function gitRunnerModuleMock(mocks: GitHubClientMocks): GitRunnerModuleMo
 }
 
 export type SshGitDispatchModuleMock = {
+  getSshGitProviderGeneration: Mock
   getSshGitProvider: Mock
   SSH_GIT_PROVIDER_UNAVAILABLE_MESSAGE: string
 }
 
 export function sshGitDispatchModuleMock(mocks: GitHubClientMocks): SshGitDispatchModuleMock {
   return {
+    getSshGitProviderGeneration: vi.fn(() => 0),
     getSshGitProvider: mocks.getSshGitProviderMock,
     SSH_GIT_PROVIDER_UNAVAILABLE_MESSAGE: 'Remote connection dropped.'
   }