diff --git a/src/models/state/mod.rs b/src/models/state/mod.rs
index 72bfcf6a..8778337f 100644
--- a/src/models/state/mod.rs
+++ b/src/models/state/mod.rs
@@ -98,12 +98,8 @@ impl GlobalState {
         let (height, time_secs) =
             time_fn_call_async(self.get_latest_balance_height_internal()).await;
 
-        #[cfg(not(test))]
         debug!("call to get_latest_balance_height() took {time_secs} seconds");
 
-        #[cfg(test)]
-        println!("call to get_latest_balance_height() took {time_secs} seconds");
-
         height
     }
 
diff --git a/src/models/state/wallet/wallet_state.rs b/src/models/state/wallet/wallet_state.rs
index 0cc9e727..6ea2663c 100644
--- a/src/models/state/wallet/wallet_state.rs
+++ b/src/models/state/wallet/wallet_state.rs
@@ -823,6 +823,7 @@ impl WalletState {
 #[cfg(test)]
 mod tests {
     use num_traits::One;
+    use tracing_test::traced_test;
 
     use crate::{
         config_models::network::Network,
@@ -833,6 +834,7 @@ mod tests {
     use super::*;
 
     #[tokio::test]
+    #[traced_test]
     async fn wallet_state_prune_abandoned_mutxos() {
         // Get genesis block. Verify wallet is empty
         // Add two blocks to state containing no UTXOs for own wallet