diff --git a/README.md b/README.md
index e2a9097..3741a2a 100644
--- a/README.md
+++ b/README.md
@@ -151,7 +151,7 @@ A function that gets initial state asynchronously without triggering actions. Op
 ###### Types
 
 ```flow
-type InitialState = () => Promise<State> | State
+type InitialState = (ctx: Context) => Promise<State> | State
 ```
 
 ---
diff --git a/src/tokens.js b/src/tokens.js
index 0765f10..7ba5f5b 100644
--- a/src/tokens.js
+++ b/src/tokens.js
@@ -13,7 +13,7 @@ import type {Token, Context} from 'fusion-core';
 
 import type {ReactReduxServiceType} from './types.js';
 
-type InitialStateType<TState> = (ctx?: Context) => Promise<TState> | TState;
+type InitialStateType<TState> = (ctx: Context) => Promise<TState> | TState;
 
 export const ReduxToken: Token<ReactReduxServiceType> = createToken(
   'ReduxToken'