diff --git a/APP_Framework/Applications/benchmark/support/benchmark.c b/APP_Framework/Applications/benchmark/support/benchmark.c
index 4046370b0..3a940bf1a 100644
--- a/APP_Framework/Applications/benchmark/support/benchmark.c
+++ b/APP_Framework/Applications/benchmark/support/benchmark.c
@@ -132,10 +132,11 @@ int run_benchmark() {
   return (!correct);
 }
 
-int main() {
+int embenchmark() {
   run_benchmark();
   printf("[BENCHMARK] Global Time Diff: %f\n", global_diff);
   return 0;
 }
+PRIV_SHELL_CMD_FUNCTION(embenchmark, run embenchmark, PRIV_SHELL_CMD_MAIN_ATTR);
 
 #endif