2 files changed, 8 insertions, 4 deletions

diff --git a/crates/mbe/src/lib.rs b/crates/mbe/src/lib.rs
index b7aa64713..b95374b76 100644
--- a/crates/mbe/src/lib.rs
+++ b/crates/mbe/src/lib.rs
@@ -64,11 +64,13 @@ impl fmt::Display for ExpandError {
     }
 }
 
-pub use crate::syntax_bridge::{
-    ast_to_token_tree, parse_exprs_with_sep, parse_to_token_tree, syntax_node_to_token_tree,
-    token_tree_to_syntax_node,
+pub use crate::{
+    syntax_bridge::{
+        ast_to_token_tree, parse_exprs_with_sep, parse_to_token_tree, syntax_node_to_token_tree,
+        token_tree_to_syntax_node,
+    },
+    token_map::TokenMap,
 };
-pub use crate::token_map::TokenMap;
 
 /// This struct contains AST for a single `macro_rules` definition. What might
 /// be very confusing is that AST has almost exactly the same shape as
diff --git a/crates/mbe/src/token_map.rs b/crates/mbe/src/token_map.rs
index 0567475be..6df3de3b3 100644
--- a/crates/mbe/src/token_map.rs
+++ b/crates/mbe/src/token_map.rs
@@ -1,3 +1,5 @@
+//! Mapping between `TokenId`s and the token's position in macro definitions or inputs.
+
 use parser::{SyntaxKind, T};
 use syntax::{TextRange, TextSize};