From 5db660a58c07be223f3b998a55677c96d402ca09 Mon Sep 17 00:00:00 2001
From: Alex Duan <417921451@qq.com>
Date: Fri, 21 Mar 2025 13:15:24 +0800
Subject: [PATCH] fix: build error

---
 tools/shell/src/shellMain.c | 2 +-
 tools/src/pub.c             | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/tools/shell/src/shellMain.c b/tools/shell/src/shellMain.c
index fa03677824..31fad97bd4 100644
--- a/tools/shell/src/shellMain.c
+++ b/tools/shell/src/shellMain.c
@@ -101,7 +101,7 @@ int main(int argc, char *argv[]) {
     return -1;
   }
 
-  if (setConnMode(&shell.args.connMode, shell.args.dsn)) {
+  if (setConnMode(shell.args.connMode, shell.args.dsn)) {
     return -1;
   }
 
diff --git a/tools/src/pub.c b/tools/src/pub.c
index 0eddbfbe04..7e561b57d9 100644
--- a/tools/src/pub.c
+++ b/tools/src/pub.c
@@ -105,7 +105,7 @@ int32_t setConnMode(int8_t connMode, char *dsn) {
     // set conn mode
     char * strMode = connMode == CONN_MODE_NATIVE ? STR_NATIVE : STR_WEBSOCKET;
     int32_t code = taos_options(TSDB_OPTION_DRIVER, strMode);
-    if (code != TSDB_CODE_SUCCESS) {
+    if (code != 0) {
       fprintf(stderr, "failed to load driver. since %s [0x%08X]\r\n", taos_errstr(NULL), taos_errno(NULL));
       return code;
     }