diff --git a/__tests__/components/shared/sprint-switcher.test.tsx b/__tests__/components/shared/sprint-switcher.test.tsx
index 8b1e2ec..ecbc6af 100644
--- a/__tests__/components/shared/sprint-switcher.test.tsx
+++ b/__tests__/components/shared/sprint-switcher.test.tsx
@@ -15,6 +15,8 @@ vi.mock('next/navigation', () => ({
 
 vi.mock('@/actions/active-sprint', () => ({
   setActiveSprintAction: vi.fn(),
+  switchActiveSprintAction: vi.fn(),
+  clearActiveSprintAction: vi.fn(),
 }))
 
 vi.mock('sonner', () => ({
@@ -23,8 +25,8 @@ vi.mock('sonner', () => ({
 
 const isDemoMock = { value: false }
 vi.mock('@/stores/user-settings/store', () => ({
-  useUserSettingsStore: (selector: (s: { context: { isDemo: boolean } }) => unknown) =>
-    selector({ context: { isDemo: isDemoMock.value } }),
+  useUserSettingsStore: (selector: (s: { context: { isDemo: boolean }; entities: { settings: { workflow: null } } }) => unknown) =>
+    selector({ context: { isDemo: isDemoMock.value }, entities: { settings: { workflow: null } } }),
 }))
 
 vi.mock('@/components/ui/dropdown-menu', () => {
@@ -54,11 +56,11 @@ vi.mock('@/components/ui/tooltip', () => {
   }
 })
 
-import { setActiveSprintAction } from '@/actions/active-sprint'
+import { switchActiveSprintAction } from '@/actions/active-sprint'
 import { toast } from 'sonner'
 import { SprintSwitcher } from '@/components/shared/sprint-switcher'
 
-const actionMock = setActiveSprintAction as unknown as ReturnType<typeof vi.fn>
+const actionMock = switchActiveSprintAction as unknown as ReturnType<typeof vi.fn>
 const toastError = toast.error as unknown as ReturnType<typeof vi.fn>
 const toastSuccess = toast.success as unknown as ReturnType<typeof vi.fn>