1 files changed, 4 insertions, 3 deletions
diff --git a/src/bin/sp-docgen.rs b/src/bin/sp-docgen.rs
index 61522ce..d633791 100644
--- a/src/bin/sp-docgen.rs
+++ b/src/bin/sp-docgen.rs
@@ -8,7 +8,9 @@ use structopt::StructOpt;
 
 use anyhow::Result;
 
-use subplot::{get_basedir_from, resource, Document, Style};
+use subplot::{resource, Style};
+
+mod cli;
 
 // Define the command line arguments.
 #[derive(Debug, StructOpt)]
@@ -42,8 +44,7 @@ fn main() -> Result<()> {
         style.typeset_links_as_notes();
     }
 
-    let basedir = get_basedir_from(first_file)?;
-    let mut doc = Document::from_file(&basedir, &first_file, style)?;
+    let mut doc = cli::load_document(&first_file, style)?;
     doc.lint()?;
     if !doc.check_named_files_exist()? {
         eprintln!("Continuing despite warnings");