diff --git a/reflex/app.py b/reflex/app.py
index fcc6c0828..62e6c73dd 100644
--- a/reflex/app.py
+++ b/reflex/app.py
@@ -270,7 +270,7 @@ class App(MiddlewareMixin, LifespanMixin):
     _state_manager: Optional[StateManager] = None
 
     # Mapping from a route to event handlers to trigger when the page loads. PRIVATE.
-    load_events: Dict[str, List[IndividualEventType[[...]]]] = dataclasses.field(
+    load_events: Dict[str, List[IndividualEventType[[]]]] = dataclasses.field(
         default_factory=dict
     )
 
diff --git a/tests/units/test_app.py b/tests/units/test_app.py
index b84955d8a..1e34a67c3 100644
--- a/tests/units/test_app.py
+++ b/tests/units/test_app.py
@@ -1211,7 +1211,7 @@ async def test_process_events(mocker, token: str):
     ],
 )
 def test_overlay_component(
-    state: State | None,
+    state: Type[State] | None,
     overlay_component: Component | ComponentCallable | None,
     exp_page_child: Type[Component] | None,
 ):
diff --git a/tests/units/test_state.py b/tests/units/test_state.py
index 8397954cf..252e26bb4 100644
--- a/tests/units/test_state.py
+++ b/tests/units/test_state.py
@@ -2724,6 +2724,7 @@ class OnLoadState(State):
 
     num: int = 0
 
+    @rx.event
     def test_handler(self):
         """Test handler."""
         self.num += 1