1 files changed, 8 insertions, 9 deletions
diff --git a/src/app/treebench/main.cc b/src/app/treebench/main.cc
index e2ddfe3..fc8ffc5 100644
--- a/src/app/treebench/main.cc
+++ b/src/app/treebench/main.cc
@@ -11,22 +11,21 @@
 
 #include "tree.c.inc"
 
-void loop(void)
+int main(void)
 {
+	arch.setup();
+	gpio.setup();
+	kout.setup();
+
 	counter.start();
 	counter.stop();
 	kout << "nop @ " << counter.value << "/" << counter.overflow << " cycles" << endl << endl;
 
 	run_benchmark();
-}
-
-int main(void)
-{
-	arch.setup();
-	gpio.setup();
-	kout.setup();
 
-	arch.idle_loop();
+	while (1) {
+		arch.idle();
+	}
 
 	return 0;
 }