diff --git a/src/main.rs b/src/main.rs
index bebbbf9..a994663 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -38,12 +38,19 @@ fn Home() -> Element {
     }
 }
 
-fn main() {
+fn tracing_setup() {
     #[cfg(debug_assertions)]
     dioxus::logger::init(tracing::Level::DEBUG).expect("Failed to initialize tracing logger");
+}
 
-    #[cfg(feature = "server")]
-    server::main();
-
+#[cfg(not(feature = "server"))]
+fn main() {
+    tracing_setup();
     dioxus::launch(App);
 }
+
+#[cfg(feature = "server")]
+fn main() {
+    tracing_setup();
+    server::main()
+}
diff --git a/src/server/main.rs b/src/server/main.rs
index a48e53e..a07663e 100644
--- a/src/server/main.rs
+++ b/src/server/main.rs
@@ -1,5 +1,9 @@
+use crate::App;
+
 pub fn main() {
     if let Err(e) = dotenvy::dotenv() {
         tracing::warn!("Error reading .env: {e}");
     }
+
+    dioxus::launch(App);
 }