diff --git a/src/core/prompts/__tests__/system.test.ts b/src/core/prompts/__tests__/system.test.ts
index d4edea8046..7f436740ad 100644
--- a/src/core/prompts/__tests__/system.test.ts
+++ b/src/core/prompts/__tests__/system.test.ts
@@ -65,10 +65,14 @@ jest.mock("os", () => ({
 	homedir: () => "/home/user",
 }))
 
-jest.mock("default-shell", () => "/bin/bash")
+jest.mock("default-shell", () => "/bin/zsh")
 
 jest.mock("os-name", () => () => "Linux")
 
+jest.mock("../../../utils/shell", () => ({
+	getShell: () => "/bin/zsh",
+}))
+
 // Create a mock ExtensionContext
 const mockContext = {
 	extensionPath: "/mock/extension/path",