diff --git a/src/hooks/useVaultAiGuidanceStatus.ts b/src/hooks/useVaultAiGuidanceStatus.ts
index fb78378c..ded8d74c 100644
--- a/src/hooks/useVaultAiGuidanceStatus.ts
+++ b/src/hooks/useVaultAiGuidanceStatus.ts
@@ -4,12 +4,13 @@ import { isTauri, mockInvoke } from '../mock-tauri'
 import {
   createCheckingVaultAiGuidanceStatus,
   normalizeVaultAiGuidanceStatus,
+  type VaultAiGuidanceFileState,
   type VaultAiGuidanceStatus,
 } from '../lib/vaultAiGuidance'
 
 type RawVaultAiGuidanceStatus = Partial<{
-  agents_state: string | null
-  claude_state: string | null
+  agents_state: VaultAiGuidanceFileState | null
+  claude_state: VaultAiGuidanceFileState | null
   can_restore: boolean | null
 }>