diff --git a/src/GlobalStateStore/GlobalStateStoreWrapper.tsx b/src/GlobalStateStore/GlobalStateStoreWrapper.tsx
index 528cfed..aae306b 100644
--- a/src/GlobalStateStore/GlobalStateStoreWrapper.tsx
+++ b/src/GlobalStateStore/GlobalStateStoreWrapper.tsx
@@ -25,9 +25,9 @@ interface GlobalStateStoreProviderProps {
   autoFetch?: boolean;
   children: ReactNode;
   fetchOnMount?: boolean;
-  onFetchSession?: (state: GlobalState) => Promise<Partial<GlobalState>>;
-  onLogin?: (state: GlobalState) => Promise<void>;
-  onLogout?: (state: GlobalState) => Promise<void>;
+  onFetchSession?: (state: Partial<GlobalState>) => Promise<Partial<GlobalState>>;
+  onLogin?: (state: Partial<GlobalState>) => Promise<Partial<GlobalState>>;
+  onLogout?: (state: Partial<GlobalState>) => Promise<Partial<GlobalState>>;
   program?: Partial<ProgramState>;
   throttleMs?: number;
 }