diff --git a/src/wallet/walletdb.rs b/src/wallet/walletdb.rs
index 4ca9d424a..34fc3525c 100644
--- a/src/wallet/walletdb.rs
+++ b/src/wallet/walletdb.rs
@@ -377,7 +377,8 @@ mod tests {
 
     use super::*;
     use crate::crypto::{coin::Coin, OwnCoin};
-    use crate::util::join_config_path;
+    use crate::serial::{deserialize, serialize};
+    use crate::util::{join_config_path, DrkTokenList, SolTokenList};
     use ff::PrimeField;
 
     pub fn init_db(path: &PathBuf, password: String) -> Result<()> {
@@ -452,61 +453,59 @@ mod tests {
         Ok(())
     }
 
-    //#[test]
-    //pub fn test_get_token_table() -> Result<()> {
-    //    let walletdb_path = join_config_path(&PathBuf::from("test2_wallet.db"))?;
-    //    let password: String = "darkfi".into();
-    //    let wallet = WalletDb::new(&walletdb_path, password.clone())?;
-    //    init_db(&walletdb_path, password)?;
+    #[test]
+    pub fn test_get_balances() -> Result<()> {
+        let walletdb_path = join_config_path(&PathBuf::from("test2_wallet.db"))?;
+        let password: String = "darkfi".into();
+        let wallet = WalletDb::new(&walletdb_path, password.clone())?;
+        init_db(&walletdb_path, password)?;
 
-    //    let secret: jubjub::Fr = jubjub::Fr::random(&mut OsRng);
-    //    let public = zcash_primitives::constants::SPENDING_KEY_GENERATOR * secret;
-    //    let key_public = serial::serialize(&public);
-    //    let key_private = serial::serialize(&secret);
+        let secret: jubjub::Fr = jubjub::Fr::random(&mut OsRng);
+        let public = zcash_primitives::constants::SPENDING_KEY_GENERATOR * secret;
+        let key_public = serial::serialize(&public);
+        let key_private = serial::serialize(&secret);
 
-    //    wallet.put_keypair(key_public, key_private)?;
+        wallet.put_keypair(key_public, key_private)?;
 
-    //    let asset_id = jubjub::Fr::random(&mut OsRng);
+        let asset_id = jubjub::Fr::random(&mut OsRng);
 
-    //    let note = Note {
-    //        serial: jubjub::Fr::random(&mut OsRng),
-    //        value: 110,
-    //        asset_id,
-    //        coin_blind: jubjub::Fr::random(&mut OsRng),
-    //        valcom_blind: jubjub::Fr::random(&mut OsRng),
-    //    };
+        let note = Note {
+            serial: jubjub::Fr::random(&mut OsRng),
+            value: 110,
+            asset_id,
+            coin_blind: jubjub::Fr::random(&mut OsRng),
+            valcom_blind: jubjub::Fr::random(&mut OsRng),
+        };
 
-    //    let coin = Coin::new(bls12_381::Scalar::random(&mut OsRng).to_repr());
+        let coin = Coin::new(bls12_381::Scalar::random(&mut OsRng).to_repr());
 
-    //    let mut tree = crate::crypto::merkle::CommitmentTree::empty();
-    //    tree.append(MerkleNode::from_coin(&coin))?;
+        let mut tree = crate::crypto::merkle::CommitmentTree::empty();
+        tree.append(MerkleNode::from_coin(&coin))?;
 
-    //    let witness = IncrementalWitness::from_tree(&tree);
+        let witness = IncrementalWitness::from_tree(&tree);
 
-    //    let own_coin = OwnCoin {
-    //        coin,
-    //        note: note.clone(),
-    //        secret,
-    //        witness: witness.clone(),
-    //    };
+        let own_coin = OwnCoin {
+            coin,
+            note: note.clone(),
+            secret,
+            witness: witness.clone(),
+        };
 
-    //    wallet.put_own_coins(own_coin.clone())?;
-    //    wallet.put_own_coins(own_coin.clone())?;
-    //    wallet.put_own_coins(own_coin.clone())?;
-    //    wallet.put_own_coins(own_coin.clone())?;
+        wallet.put_own_coins(own_coin.clone())?;
+        wallet.put_own_coins(own_coin.clone())?;
+        wallet.put_own_coins(own_coin.clone())?;
+        wallet.put_own_coins(own_coin.clone())?;
 
-    //    let table_vec = wallet.get_balances()?;
+        let balances = wallet.get_balances()?;
 
-    //    assert_eq!(table_vec.len(), 4);
-    //    assert_eq!(table_vec[0].value, 110);
-    //    assert_eq!(table_vec[0].token_id, asset_id);
-    //    assert_eq!(table_vec[2].value, 110);
-    //    assert_eq!(table_vec[2].token_id, asset_id);
+        let asset_id = serialize(&asset_id);
 
-    //    std::fs::remove_file(walletdb_path)?;
+        assert_eq!(balances[&asset_id], 110);
 
-    //    Ok(())
-    //}
+        std::fs::remove_file(walletdb_path)?;
+
+        Ok(())
+    }
 
     #[test]
     pub fn test_save_and_load_keypair() -> Result<()> {