diff --git a/at_socket_esp8266.c b/at_socket_esp8266.c
index 6678e85..63575ec 100644
--- a/at_socket_esp8266.c
+++ b/at_socket_esp8266.c
@@ -240,7 +240,6 @@ static int esp8266_socket_send(int socket, const char *buff, size_t bfsz, enum a
     /* set current socket for send URC event */
     cur_socket = socket;
     /* set AT client end sign to deal with '>' sign.*/
-    extern int at_set_end_sign(char ch);
     at_set_end_sign('>');
 
     while (sent_size < bfsz)
@@ -602,7 +601,7 @@ __exit:
 
 int esp8266_net_init(void)
 {
-#ifdef PKG_AT_INIT_BY_THREAD
+#ifdef AT_DEVICE_INIT_BY_THREAD
     rt_thread_t tid;
 
     tid = rt_thread_create("esp8266_net_init", esp8266_init_thread_entry, RT_NULL,ESP8266_THREAD_STACK_SIZE, ESP8266_THREAD_PRIORITY, 20);
@@ -702,7 +701,7 @@ static int at_socket_device_init(void)
     }
 
     /* initialize AT client */
-    at_client_init();
+    at_client_init(AT_DEVICE_NAME, AT_DEVICE_RECV_BUFF_LEN);
 
     /* register URC data execution function  */
     at_set_urc_table(urc_table, sizeof(urc_table) / sizeof(urc_table[0]));
diff --git a/at_socket_m26.c b/at_socket_m26.c
index 3924afc..20f98fa 100644
--- a/at_socket_m26.c
+++ b/at_socket_m26.c
@@ -293,7 +293,6 @@ static int m26_socket_send(int socket, const char *buff, size_t bfsz, enum at_so
     /* set current socket for send URC event */
     cur_socket = socket;
     /* set AT client end sign to deal with '>' sign.*/
-    extern int at_set_end_sign(char ch);
     at_set_end_sign('>');
 
     while (sent_size < bfsz)
@@ -781,7 +780,7 @@ __exit:
 
 int m26_net_init(void)
 {
-#ifdef PKG_AT_INIT_BY_THREAD
+#ifdef AT_DEVICE_INIT_BY_THREAD
     rt_thread_t tid;
 
     tid = rt_thread_create("m26_net_init", m26_init_thread_entry, RT_NULL, M26_THREAD_STACK_SIZE, M26_THREAD_PRIORITY, 20);
@@ -907,7 +906,7 @@ static int at_socket_device_init(void)
     }
 
     /* initialize AT client */
-    at_client_init();
+    at_client_init(AT_DEVICE_NAME, AT_DEVICE_RECV_BUFF_LEN);
 
     /* register URC data execution function  */
     at_set_urc_table(urc_table, sizeof(urc_table) / sizeof(urc_table[0]));