diff --git a/common/src/backends/metrics.rs b/common/src/backends/metrics.rs
index 9d7fc2b778..0ecc7ebec0 100644
--- a/common/src/backends/metrics.rs
+++ b/common/src/backends/metrics.rs
@@ -54,6 +54,9 @@ impl<MakeSpan> TraceLayer<MakeSpan> {
     ///
     /// # Example
     /// ```
+    /// use shuttle_common::{request_span, backends::metrics::TraceLayer};
+    /// use tracing::field;
+    ///
     /// TraceLayer::new(|request| {
     ///     request_span!(
     ///         request,
@@ -206,7 +209,7 @@ mod tests {
     use tower::ServiceExt;
     use tracing::field;
     use tracing_fluent_assertions::{AssertionRegistry, AssertionsLayer};
-    use tracing_subscriber::{layer::SubscriberExt, Registry};
+    use tracing_subscriber::layer::SubscriberExt;
 
     use super::{Metrics, TraceLayer};
 
@@ -221,9 +224,9 @@ mod tests {
     #[tokio::test]
     async fn trace_layer() {
         let assertion_registry = AssertionRegistry::default();
-        let base_subscriber = Registry::default();
-        let subscriber = base_subscriber.with(AssertionsLayer::new(&assertion_registry));
-        tracing::subscriber::set_global_default(subscriber).unwrap();
+        let subscriber =
+            tracing_subscriber::registry().with(AssertionsLayer::new(&assertion_registry));
+        let _guard = tracing::subscriber::set_default(subscriber);
 
         // Put in own block to make sure assertion to not interfere with the next test
         {
diff --git a/common/src/wasm.rs b/common/src/wasm.rs
index edaa8808b3..1508644dd2 100644
--- a/common/src/wasm.rs
+++ b/common/src/wasm.rs
@@ -366,7 +366,7 @@ where
 }
 
 #[cfg(test)]
-mod test {
+mod tests {
     use cap_std::os::unix::net::UnixStream;
     use serde_json::json;
     use std::io::{Read, Write};
@@ -499,7 +499,7 @@ mod test {
             (message, log.level)
         };
 
-        tracing_subscriber::registry().with(logger).init();
+        let _guard = tracing_subscriber::registry().with(logger).set_default();
 
         tracing::debug!("this is");
         tracing::info!("hi");
diff --git a/runtime/src/logger.rs b/runtime/src/logger.rs
index 79b2f6a496..8a82c2508d 100644
--- a/runtime/src/logger.rs
+++ b/runtime/src/logger.rs
@@ -64,7 +64,7 @@ mod tests {
 
         let logger = Logger::new(s, Default::default());
 
-        tracing_subscriber::registry().with(logger).init();
+        let _guard = tracing_subscriber::registry().with(logger).set_default();
 
         tracing::debug!("this is");
         tracing::info!("hi");