diff --git a/tools/shell/inc/shellAuto.h b/tools/shell/inc/shellAuto.h
index b7bf5fa1019502acbeaefc8884d4553704f58702..151f6da8c6555118699eedbf791da32e47e4283e 100644
--- a/tools/shell/inc/shellAuto.h
+++ b/tools/shell/inc/shellAuto.h
@@ -28,7 +28,7 @@ void pressOtherKey(char c);
 bool shellAutoInit();
 
 // set conn
-void shellSetConn(TAOS* conn);
+void shellSetConn(TAOS* conn, bool runOnce);
 
 // exit shell auto funciton, shell exit call once
 void shellAutoExit();