diff --git a/crates/tuono_lib/src/internal_handlers/catch_all.rs b/crates/tuono_lib/src/catch_all.rs
similarity index 100%
rename from crates/tuono_lib/src/internal_handlers/catch_all.rs
rename to crates/tuono_lib/src/catch_all.rs
diff --git a/crates/tuono_lib/src/internal_handlers.rs b/crates/tuono_lib/src/internal_handlers.rs
deleted file mode 100644
index 02857958..00000000
--- a/crates/tuono_lib/src/internal_handlers.rs
+++ /dev/null
@@ -1,7 +0,0 @@
-mod catch_all;
-mod vite_reverse_proxy;
-mod vite_websocket_proxy;
-
-pub use catch_all::catch_all;
-pub use vite_reverse_proxy::vite_reverse_proxy;
-pub use vite_websocket_proxy::vite_websocket_proxy;
diff --git a/crates/tuono_lib/src/lib.rs b/crates/tuono_lib/src/lib.rs
index 15283098..067533ed 100644
--- a/crates/tuono_lib/src/lib.rs
+++ b/crates/tuono_lib/src/lib.rs
@@ -1,4 +1,4 @@
-mod internal_handlers;
+mod catch_all;
 mod manifest;
 mod mode;
 mod payload;
@@ -6,6 +6,8 @@ mod request;
 mod response;
 mod server;
 mod ssr;
+mod vite_reverse_proxy;
+mod vite_websocket_proxy;
 
 pub use mode::Mode;
 pub use payload::Payload;
diff --git a/crates/tuono_lib/src/server.rs b/crates/tuono_lib/src/server.rs
index c64fa922..f3e00632 100644
--- a/crates/tuono_lib/src/server.rs
+++ b/crates/tuono_lib/src/server.rs
@@ -5,7 +5,10 @@ use axum::routing::{get, Router};
 use ssr_rs::Ssr;
 use tower_http::services::ServeDir;
 
-use crate::internal_handlers::{catch_all, vite_reverse_proxy, vite_websocket_proxy};
+use crate::{
+    catch_all::catch_all, vite_reverse_proxy::vite_reverse_proxy,
+    vite_websocket_proxy::vite_websocket_proxy,
+};
 
 const DEV_PUBLIC_DIR: &str = "public";
 const PROD_PUBLIC_DIR: &str = "out/client";
diff --git a/crates/tuono_lib/src/internal_handlers/vite_reverse_proxy.rs b/crates/tuono_lib/src/vite_reverse_proxy.rs
similarity index 100%
rename from crates/tuono_lib/src/internal_handlers/vite_reverse_proxy.rs
rename to crates/tuono_lib/src/vite_reverse_proxy.rs
diff --git a/crates/tuono_lib/src/internal_handlers/vite_websocket_proxy.rs b/crates/tuono_lib/src/vite_websocket_proxy.rs
similarity index 100%
rename from crates/tuono_lib/src/internal_handlers/vite_websocket_proxy.rs
rename to crates/tuono_lib/src/vite_websocket_proxy.rs