diff --git a/pynecone/middleware/hydrate_middleware.py b/pynecone/middleware/hydrate_middleware.py
index 111b3c31b..5d933e5c0 100644
--- a/pynecone/middleware/hydrate_middleware.py
+++ b/pynecone/middleware/hydrate_middleware.py
@@ -35,5 +35,7 @@ class HydrateMiddleware(Middleware):
             else:
                 load_event = None
             if load_event:
-                load_event.fn(state)
+                substate_path = utils.format_event_handler(load_event).split(".")
+                ex_state = state.get_substate(substate_path[:-1])
+                load_event.fn(ex_state)
             return utils.format_state({state.get_name(): state.dict()})