diff --git a/src/benchmark.py b/src/benchmark.py
--- a/src/benchmark.py
+++ b/src/benchmark.py
@@ -1,10 +1,14 @@
 import cProfile
 
 from tests.testEngine import TestTransitions
+from tests.testStatebag import TestStateBag
 
 
-t=TestTransitions()
+# t=TestTransitions()
+#
+# cProfile.run(r"""
+# t.testReal()
+# """)
 
-cProfile.run(r"""
-t.testReal()
-""")
+t=TestStateBag()
+cProfile.run(r"""t.testReal()""")