diff --git a/router/src/server.rs b/router/src/server.rs
index db6c369..623dd07 100644
--- a/router/src/server.rs
+++ b/router/src/server.rs
@@ -195,8 +195,8 @@ pub async fn run(
 
     // Create router
     let app = Router::new()
+        .route("/", post(generate))
         .route("/generate", post(generate))
-        .layer(Extension(shared_state.clone()))
         .route("/", get(health))
         .route("/health", get(health))
         .layer(Extension(shared_state.clone()));