diff --git a/src/main.rs b/src/main.rs
index a034646..a39ae54 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,13 +1,11 @@
-use liblambdashell::shell;
-
 mod cli;
 
 fn main() {
 	if let Some(args) = cli::parser() {
-		let shell_config = shell::Config {
+		let shell_config = liblambdashell::session::Config {
 			norc: args.norc,
 		};
-		let mut shell_instance = shell::LambdaShell::create(shell_config);
+		let mut shell_instance = liblambdashell::session::LambdaShell::create(shell_config);
 		shell_instance.start();
 	};
 }
\ No newline at end of file