diff --git a/src/main.rs b/src/main.rs
index 2358a72..4c4fcc3 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -2,21 +2,12 @@
 //
 // SPDX-License-Identifier: AGPL-3.0-only OR LicenseRef-Commercial
 
+#![forbid(unsafe_code)]
+
 use clap::Parser;
+use commitbee::{App, Cli, error};
 use tracing_subscriber::EnvFilter;
 
-mod app;
-mod cli;
-mod config;
-mod domain;
-mod error;
-#[cfg(feature = "eval")]
-mod eval;
-mod services;
-
-use app::App;
-use cli::Cli;
-
 #[tokio::main]
 async fn main() {
     miette::set_hook(Box::new(|_| {