1 files changed, 4 insertions, 4 deletions
diff --git a/src/lib.rs b/src/lib.rs
index 3483847..69be41f 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -9,10 +9,12 @@ use chain::{atom::DefaultAtom, default::DefaultChain, Chain};
 use grammar::Grammar;
 
 // For printing forests
-use chain::item::{default::DefaultForest, ForestLabel, ForestLabelType};
+use chain::item::default::DefaultForest;
 use grammar::{GrammarLabel, TNT};
 use graph::LabelGraph;
 
+use forest::{ForestLabel, ForestLabelType};
+
 /// This struct is the representation of a parser.
 ///
 /// When the user constructs a parser, an instance of this struct will
@@ -485,7 +487,7 @@ extern "C" fn parser_parse(
                     Ok(forest) => {
                         Box::leak(parser_box);
 
-                        let mut bytes = bytes::forest_to_bytes(&forest);
+                        let mut bytes = forest::forest_to_bytes(&forest);
 
                         let bytes_len = bytes.len().to_be_bytes().to_vec();
 
@@ -835,5 +837,3 @@ extern "C" fn print_forest_node(forest: *mut CForest, node: *mut std::os::raw::c
 
     println!("node {node} has label {label}");
 }
-
-pub mod bytes;