diff --git a/src/db/app.rs b/src/db/app.rs
index f5d3f63..77de30c 100644
--- a/src/db/app.rs
+++ b/src/db/app.rs
@@ -6,9 +6,8 @@ use super::Error;
 use crate::constants::{
     ACCOUNT, ACTIVE_APP, APP_DAEMON_TIMEOUT, APP_NODE, DEFAULT_ENDPOINT, DELETED_APP, NEW_APP_REQ,
 };
-use crate::db;
 use crate::db::general::Report;
-use crate::old_brain;
+use crate::{db, old_brain};
 use detee_shared::app_proto::{self, NewAppRes};
 use serde::{Deserialize, Serialize};
 use surrealdb::engine::remote::ws::Client;
diff --git a/src/grpc/app.rs b/src/grpc/app.rs
index 0d7087c..b268cc4 100644
--- a/src/grpc/app.rs
+++ b/src/grpc/app.rs
@@ -12,8 +12,7 @@ use log::info;
 use std::pin::Pin;
 use std::sync::Arc;
 use surrealdb::engine::remote::ws::Client;
-use surrealdb::RecordId;
-use surrealdb::Surreal;
+use surrealdb::{RecordId, Surreal};
 use tokio::sync::mpsc;
 use tokio_stream::wrappers::ReceiverStream;
 use tokio_stream::{Stream, StreamExt};
@@ -31,8 +30,8 @@ impl AppDaemonServer {
 
 #[tonic::async_trait]
 impl BrainAppDaemon for AppDaemonServer {
-    type RegisterAppNodeStream = Pin<Box<dyn Stream<Item = Result<DelAppReq, Status>> + Send>>;
     type BrainMessagesStream = Pin<Box<dyn Stream<Item = Result<BrainMessageApp, Status>> + Send>>;
+    type RegisterAppNodeStream = Pin<Box<dyn Stream<Item = Result<DelAppReq, Status>> + Send>>;
 
     async fn register_app_node(
         &self,
diff --git a/src/grpc/types.rs b/src/grpc/types.rs
index 11b719e..25c0240 100644
--- a/src/grpc/types.rs
+++ b/src/grpc/types.rs
@@ -2,10 +2,10 @@
 
 use crate::constants::{ACCOUNT, APP_NODE, ID_ALPHABET, NEW_APP_REQ, NEW_VM_REQ, VM_NODE};
 use crate::db::prelude as db;
-use detee_shared::app_proto::AppNodeListResp;
+use detee_shared::app_proto::*;
 use detee_shared::common_proto::MappedPort;
 use detee_shared::general_proto::{Account, AccountBalance, ListOperatorsResp};
-use detee_shared::{app_proto::*, vm_proto::*};
+use detee_shared::vm_proto::*;
 use nanoid::nanoid;
 
 use surrealdb::RecordId;
diff --git a/tests/common/app_cli_utils.rs b/tests/common/app_cli_utils.rs
index db05cb7..7487571 100644
--- a/tests/common/app_cli_utils.rs
+++ b/tests/common/app_cli_utils.rs
@@ -1,9 +1,8 @@
 // SPDX-License-Identifier: Apache-2.0
 
 use anyhow::Result;
-use detee_shared::app_proto::{
-    brain_app_cli_client::BrainAppCliClient, AppResource, NewAppReq, NewAppRes,
-};
+use detee_shared::app_proto::brain_app_cli_client::BrainAppCliClient;
+use detee_shared::app_proto::{AppResource, NewAppReq, NewAppRes};
 use tonic::transport::Channel;
 
 use crate::common::test_utils::Key;
diff --git a/tests/common/test_utils.rs b/tests/common/test_utils.rs
index 4aece2d..1c1e14d 100644
--- a/tests/common/test_utils.rs
+++ b/tests/common/test_utils.rs
@@ -1,10 +1,9 @@
 // SPDX-License-Identifier: Apache-2.0
 
 use anyhow::Result;
-use detee_shared::app_proto as sgx_proto;
 use detee_shared::general_proto::brain_general_cli_client::BrainGeneralCliClient;
 use detee_shared::general_proto::AirdropReq;
-use detee_shared::vm_proto as snp_proto;
+use detee_shared::{app_proto, vm_proto};
 use ed25519_dalek::{Signer, SigningKey};
 use itertools::Itertools;
 use rand::Rng;
@@ -75,20 +74,20 @@ impl Key {
     pub fn sign_stream_auth_vm(
         &self,
         contracts: Vec<String>,
-    ) -> Result<snp_proto::DaemonStreamAuth> {
+    ) -> Result<vm_proto::DaemonStreamAuth> {
         let pubkey = self.pubkey.clone();
         let timestamp = chrono::Utc::now().to_rfc3339();
         let signature =
             self.try_sign_message(&(timestamp.to_string() + &format!("{contracts:?}")))?;
-        Ok(snp_proto::DaemonStreamAuth { timestamp, pubkey, contracts, signature })
+        Ok(vm_proto::DaemonStreamAuth { timestamp, pubkey, contracts, signature })
     }
 
-    pub fn sign_stream_auth_app(&self, contracts: Vec<String>) -> Result<sgx_proto::DaemonAuth> {
+    pub fn sign_stream_auth_app(&self, contracts: Vec<String>) -> Result<app_proto::DaemonAuth> {
         let pubkey = self.pubkey.clone();
         let timestamp = chrono::Utc::now().to_rfc3339();
         let signature =
             self.try_sign_message(&(timestamp.to_string() + &format!("{contracts:?}")))?;
-        Ok(sgx_proto::DaemonAuth { timestamp, pubkey, contracts, signature })
+        Ok(app_proto::DaemonAuth { timestamp, pubkey, contracts, signature })
     }
 }
 
diff --git a/tests/common/vm_cli_utils.rs b/tests/common/vm_cli_utils.rs
index 7db9d54..64ba998 100644
--- a/tests/common/vm_cli_utils.rs
+++ b/tests/common/vm_cli_utils.rs
@@ -2,12 +2,11 @@
 
 use super::test_utils::Key;
 use anyhow::{anyhow, Result};
-use detee_shared::app_proto;
 use detee_shared::common_proto::Empty;
 use detee_shared::general_proto::brain_general_cli_client::BrainGeneralCliClient;
 use detee_shared::general_proto::{Account, RegOperatorReq, ReportNodeReq};
-use detee_shared::vm_proto;
 use detee_shared::vm_proto::brain_vm_cli_client::BrainVmCliClient;
+use detee_shared::{app_proto, vm_proto};
 use futures::StreamExt;
 use surreal_brain::constants::{ACTIVE_VM, NEW_VM_REQ};
 use surreal_brain::db::prelude as db;
diff --git a/tests/grpc_redirect_test.rs b/tests/grpc_redirect_test.rs
index 0573b35..320efdb 100644
--- a/tests/grpc_redirect_test.rs
+++ b/tests/grpc_redirect_test.rs
@@ -3,12 +3,11 @@
 use common::prepare_test_env::{prepare_test_db, run_service_for_stream};
 use common::test_utils::{airdrop, Key};
 use common::vm_daemon_utils::register_vm_node;
-use detee_shared::app_proto;
 use detee_shared::app_proto::brain_app_cli_client::BrainAppCliClient;
 use detee_shared::app_proto::brain_app_daemon_client::BrainAppDaemonClient;
-use detee_shared::vm_proto;
 use detee_shared::vm_proto::brain_vm_cli_client::BrainVmCliClient;
 use detee_shared::vm_proto::brain_vm_daemon_client::BrainVmDaemonClient;
+use detee_shared::{app_proto, vm_proto};
 
 use crate::common::app_daemon_utils::register_app_node;