diff --git a/src/benchmark.py b/src/benchmark.py
--- a/src/benchmark.py
+++ b/src/benchmark.py
@@ -1,14 +1,15 @@
+import sys
 import cProfile
 
 from tests.testEngine import TestTransitions
 from tests.testStatebag import TestStateBag
 
 
-# t=TestTransitions()
-#
-# cProfile.run(r"""
-# t.testReal()
-# """)
+arg=sys.argv[1]
 
-t=TestStateBag()
-cProfile.run(r"""t.testReal()""")
+if arg=="engine":
+	t=TestTransitions()
+	cProfile.run(r"""t.testReal()""")
+elif arg=="statebag":
+	t=TestStateBag()
+	cProfile.run(r"""t.testReal()""")