diff --git a/Cargo.lock b/Cargo.lock
index af66353..606d9b0 100644
--- a/Cargo.lock
+++ b/Cargo.lock
@@ -373,7 +373,7 @@ dependencies = [
 [[package]]
 name = "detee-shared"
 version = "0.1.0"
-source = "git+ssh://git@gitea.detee.cloud/testnet/proto?branch=main#be4e41db050c6d59e9fb5abf47e647f5bbdc24b2"
+source = "git+ssh://git@gitea.detee.cloud/testnet/proto?branch=main#cf0c9a2c0d2edf9254f25c6faa7494afcfa00d64"
 dependencies = [
  "base64",
  "prost",
diff --git a/src/grpc.rs b/src/grpc.rs
index 6ea92e2..2b1d18d 100644
--- a/src/grpc.rs
+++ b/src/grpc.rs
@@ -2,7 +2,9 @@ use crate::global::*;
 use crate::snp_proto::VmDaemonMessage;
 use anyhow::Result;
 use log::{debug, info, warn};
-use snp_proto::{brain_vm_daemon_client::BrainVmDaemonClient, BrainVmMessage, VmContract, RegisterVmNodeReq};
+use snp_proto::{
+    brain_vm_daemon_client::BrainVmDaemonClient, BrainVmMessage, RegisterVmNodeReq, VmContract,
+};
 use tokio::{
     sync::mpsc::{Receiver, Sender},
     task::JoinSet,
@@ -11,8 +13,7 @@ use tokio_stream::{wrappers::ReceiverStream, StreamExt};
 use tonic::transport::Channel;
 
 pub mod snp_proto {
-    // tonic::include_proto!("vm_proto");
-    pub use detee_shared::snp::pb::vm::*;
+    pub use detee_shared::vm_proto::*;
 }
 
 pub async fn register_node(config: &crate::config::Config) -> Result<Vec<VmContract>> {