diff --git a/src/lib.rs b/src/lib.rs
index 1de6955..ea956cb 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -2,7 +2,14 @@ mod account;
 mod db;
 mod model;
 
-pub fn router() -> axum::Router {
+pub async fn run() {
+    let app = router();
+    let listener = tokio::net::TcpListener::bind("0.0.0.0:3000").await.unwrap();
+
+    axum::serve(listener, app).await.unwrap();
+}
+
+fn router() -> axum::Router {
     axum::Router::new()
         .nest("/account", account::router())
         .nest("/predict", model::router())
diff --git a/src/main.rs b/src/main.rs
index 9e88f72..272213f 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,8 +1,4 @@
 #[tokio::main]
 async fn main() {
-    let app = dermy_server::api::api_router();
-    let listener = tokio::net::TcpListener::bind("0.0.0.0:3000").await.unwrap();
-
-    axum::serve(listener, app).await.unwrap();
-
+    dermy_server::run().await;
 }