diff --git a/site/src/types/onboarding.ts b/site/src/types/onboarding.ts
index ab8b65a..c94cc84 100644
--- a/site/src/types/onboarding.ts
+++ b/site/src/types/onboarding.ts
@@ -35,11 +35,11 @@ export type OnboardingFlow = {
 
 export type OnboardingSession = {
     uuid: string
-    flow: string
-    agent_run?: string | null
-    status: 'in_progress' | 'completed' | 'abandoned'
-    current_page_order: number
-    responses: Record<string, unknown>
+    status: OnboardingSessionStatus | string
+    role?: string | UuidNameRef
+    state?: Record<string, unknown>
+    active_configs?: Record<string, unknown>
+    completed_at?: string | null
 }
 
 export type OnboardingFeedback = {