diff --git a/tests/integration/test_component_state.py b/tests/integration/test_component_state.py
index 71e455f16..f4a295d07 100644
--- a/tests/integration/test_component_state.py
+++ b/tests/integration/test_component_state.py
@@ -58,6 +58,7 @@ def ComponentStateApp():
         class _Counter(rx.State):
             count: int = 0
 
+            @rx.event
             def increment(self):
                 self.count += 1