diff --git a/src/configure.rs b/src/configure.rs
index ac9b93e3..bb70c88c 100644
--- a/src/configure.rs
+++ b/src/configure.rs
@@ -4,7 +4,7 @@ use std::io::ErrorKind;
 use std::process;
 use std::process::Command;
 
-use starship::config::StarshipConfig;
+use crate::config::StarshipConfig;
 use std::fs::File;
 use std::io::Write;
 use toml::map::Map;
diff --git a/src/lib.rs b/src/lib.rs
index 15a38074..fb2361b8 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -1,13 +1,16 @@
 // Lib is present to allow for benchmarking
+pub mod bug_report;
 pub mod config;
 pub mod configs;
+pub mod configure;
 pub mod context;
 pub mod formatter;
+pub mod init;
 pub mod logger;
 pub mod module;
-pub mod modules;
+mod modules;
 pub mod print;
-pub mod segment;
+mod segment;
 mod utils;
 
 #[cfg(test)]
diff --git a/src/main.rs b/src/main.rs
index 46dc86e2..3957faa6 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -2,27 +2,11 @@ use clap::{crate_authors, crate_version};
 use std::io;
 use std::time::SystemTime;
 
-mod bug_report;
-mod config;
-mod configs;
-mod configure;
-mod context;
-mod formatter;
-mod init;
-mod logger;
-mod module;
-mod modules;
-mod print;
-mod segment;
-mod utils;
-
-#[cfg(test)]
-mod test;
-
-use crate::module::ALL_MODULES;
 use clap::{App, AppSettings, Arg, Shell, SubCommand};
 use rand::distributions::Alphanumeric;
 use rand::Rng;
+use starship::module::ALL_MODULES;
+use starship::*;
 
 fn main() {
     logger::init();