diff --git a/pubky/src/lib.rs b/pubky/src/lib.rs
index 339255f..eca040e 100644
--- a/pubky/src/lib.rs
+++ b/pubky/src/lib.rs
@@ -4,7 +4,7 @@
 // TODO: deny missing docs.
 // #![deny(missing_docs)]
 #![deny(rustdoc::broken_intra_doc_links)]
-// TODO: deny unwrap
+// TODO: deny unwrap only in test
 #![cfg_attr(any(), deny(clippy::unwrap_used))]
 
 macro_rules! cross_debug {
diff --git a/pubky/src/native.rs b/pubky/src/native.rs
index f988280..e9370da 100644
--- a/pubky/src/native.rs
+++ b/pubky/src/native.rs
@@ -50,7 +50,7 @@ impl ClientBuilder {
         self
     }
 
-    /// Create a [mainline::DhtBuilder] if `None`, and allows mutating it with a callback function.
+    /// Allows mutating the internal [pkarr::ClientBuilder] with a callback function.
     pub fn pkarr<F>(&mut self, f: F) -> &mut Self
     where
         F: FnOnce(&mut pkarr::ClientBuilder) -> &mut pkarr::ClientBuilder,