diff --git a/tests/benchmarks/test_compilation.py b/tests/benchmarks/test_compilation.py
index e9640d4e3..0a20ed521 100644
--- a/tests/benchmarks/test_compilation.py
+++ b/tests/benchmarks/test_compilation.py
@@ -1,6 +1,6 @@
 import pytest
 
-from reflex.compiler.compiler import _compile_page
+from reflex.compiler.compiler import _compile_page, _compile_stateful_components
 
 
 @pytest.mark.benchmark
@@ -8,6 +8,11 @@ def test_compile_page(evaluated_page):
     _compile_page(evaluated_page, None)
 
 
+@pytest.mark.benchmark
+def test_compile_stateful(evaluated_page):
+    _compile_stateful_components([evaluated_page])
+
+
 @pytest.mark.benchmark
 def test_get_all_imports(evaluated_page):
     evaluated_page._get_all_imports()