1 files changed, 11 insertions, 12 deletions

diff --git a/src/generator/mod.rs b/src/generator/mod.rs
index 3f62779..f351bb1 100644
--- a/src/generator/mod.rs
+++ b/src/generator/mod.rs
@@ -17,7 +17,7 @@ const DEFAULT_CSS: &[u8] = include_bytes!("../../res/style.css");
 
 pub(crate) fn generate(
     base_dir: &Path,
-    entities: BTreeMap<EntityId, Entity>,
+    toplevel_entity: Entity,
     config: &Config,
 ) -> Result<()> {
     let md_output_dir = base_dir.join("markdown");
@@ -38,17 +38,16 @@ pub(crate) fn generate(
     debug!("Generated temporary file with CSS at: {:?}", css_path);
 
     rayon::scope(|scope| {
-        for (id, entity) in entities {
-            generate_recursively(
-                id,
-                entity,
-                scope,
-                &md_output_dir,
-                &html_output_dir,
-                css_path,
-                config,
-            );
-        }
+        generate_recursively(
+            // TODO: a bit hacky?
+            EntityId(String::from("index")),
+            toplevel_entity,
+            scope,
+            &md_output_dir,
+            &html_output_dir,
+            css_path,
+            config,
+        );
     });
 
     Ok(())