diff --git a/packages/server/examples/axum-router/src/main.rs b/packages/server/examples/axum-router/src/main.rs
index 0b5e55e4f..692bc91af 100644
--- a/packages/server/examples/axum-router/src/main.rs
+++ b/packages/server/examples/axum-router/src/main.rs
@@ -50,6 +50,8 @@ fn main() {
                         axum::Router::new()
                             // Register server functions
                             .register_server_fns("")
+                            // Connect to the hot reload server
+                            .connect_hot_reload()
                             // Serve the static assets folder
                             .nest_service("/assets", serve_dir)
                             // If the path is unknown, render the application
diff --git a/packages/server/src/adapters/axum_adapter.rs b/packages/server/src/adapters/axum_adapter.rs
index 6ca82b1e5..b5ee3772a 100644
--- a/packages/server/src/adapters/axum_adapter.rs
+++ b/packages/server/src/adapters/axum_adapter.rs
@@ -290,15 +290,10 @@ where
             self.nest(
                 "/_dioxus",
                 Router::new()
-                    .route("/hot_reload", get(hot_reload_handler))
                     .route(
                         "/disconnect",
                         get(|ws: WebSocketUpgrade| async {
-                            ws.on_failed_upgrade(|error| {
-                                println!("failed to upgrade: {}", error);
-                                todo!()
-                            })
-                            .on_upgrade(|mut ws| async move {
+                            ws.on_upgrade(|mut ws| async move {
                                 use axum::extract::ws::Message;
                                 let _ = ws.send(Message::Text("connected".into())).await;
                                 loop {
@@ -308,7 +303,8 @@ where
                                 }
                             })
                         }),
-                    ),
+                    )
+                    .route("/hot_reload", get(hot_reload_handler)),
             )
         }
         #[cfg(not(all(debug_assertions, feature = "hot-reload", feature = "ssr")))]