diff --git a/APP_Framework/Applications/knowing_app/mnist/main.cpp b/APP_Framework/Applications/knowing_app/mnist/main.cpp
index 3f2d3321a..c8c264490 100644
--- a/APP_Framework/Applications/knowing_app/mnist/main.cpp
+++ b/APP_Framework/Applications/knowing_app/mnist/main.cpp
@@ -1,5 +1,5 @@
 #include <cstdio>
-#include <transform.h>
+// #include <transform.h>
 #include <stdio.h>
 
 #include "tensorflow/lite/micro/all_ops_resolver.h"
diff --git a/APP_Framework/Applications/knowing_app/mnist/mnistmain.c b/APP_Framework/Applications/knowing_app/mnist/mnistmain.c
index fa1c26821..b0f647d49 100644
--- a/APP_Framework/Applications/knowing_app/mnist/mnistmain.c
+++ b/APP_Framework/Applications/knowing_app/mnist/mnistmain.c
@@ -26,3 +26,4 @@ void mnist_app(void);
 int tfmnist(void) {
     mnist_app();
 }
+PRIV_SHELL_CMD_FUNCTION(tfmnist, a tenorflow_lite_for_microcontroller sample, PRIV_SHELL_CMD_FUNC_ATTR);