diff --git a/src/bin/node/miner.rs b/src/bin/node/miner.rs
index a237d5d..6417976 100644
--- a/src/bin/node/miner.rs
+++ b/src/bin/node/miner.rs
@@ -1,6 +1,7 @@
 //! App to run a mining node.
 
 use aiblock_network::configurations::{ExtraNodeParams, MinerNodeConfig, UserNodeConfig};
+use aiblock_network::wallet_db::WalletDb;
 use aiblock_network::{
     loop_wait_connnect_to_peers_async, loops_re_connect_disconnect, routes, shutdown_connections,
     ResponseResult,
@@ -18,8 +19,9 @@ pub async fn run_node(matches: &ArgMatches<'_>) {
     let node = MinerNode::new(config, Default::default()).await.unwrap();
     info!("Started node at {}", node.local_address());
 
+    let wallet_db = WalletDb::new();
     let miner_api_inputs = node.api_inputs();
-    let shared_wallet_db = Some(node.get_wallet_db().clone());
+    let shared_wallet_db = Some(wallet_db.clone());
     let (node_conn, addrs_to_connect, expected_connected_addrs) = node.connect_info_peers();
     let local_event_tx = node.local_event_tx().clone();
 
@@ -60,8 +62,10 @@ pub async fn run_node(matches: &ArgMatches<'_>) {
 
     match user_config {
         Some(config) => {
+            let user_wallet_db = WalletDb::new();
             let shared_members = ExtraNodeParams {
                 shared_wallet_db,
+                user_wallet_db,
                 ..Default::default()
             };
 
@@ -701,4 +705,4 @@ mod test {
             None => assert!(config.1.is_none()),
         }
     }
-}
+}
\ No newline at end of file
diff --git a/src/bin/node/storage.rs b/src/bin/node/storage.rs
index cceed63..10efe43 100644
--- a/src/bin/node/storage.rs
+++ b/src/bin/node/storage.rs
@@ -2,6 +2,7 @@
 
 use aiblock_network::configurations::StorageNodeConfig;
 use aiblock_network::StorageNode;
+use aiblock_network::WalletDb;
 use aiblock_network::{
     loop_wait_connnect_to_peers_async, loops_re_connect_disconnect, routes, shutdown_connections,
     ResponseResult,
@@ -15,7 +16,8 @@ pub async fn run_node(matches: &ArgMatches<'_>) {
     let config = configuration(load_settings(matches));
 
     info!("Start node with config {config:?}");
-    let node = StorageNode::new(config, Default::default()).await.unwrap();
+    let wallet_db = WalletDb::new();
+    let node = StorageNode::new(config, wallet_db).await.unwrap();
 
     info!("Started node at {}", node.local_address());
 
@@ -330,4 +332,4 @@ mod test {
             expected_key
         );
     }
-}
+}
\ No newline at end of file