1 files changed, 3 insertions, 2 deletions
diff --git a/src/server.rs b/src/server.rs
index c6ce02c..bb2b4d0 100644
--- a/src/server.rs
+++ b/src/server.rs
@@ -1,4 +1,4 @@
-use activitypub_federation::config::FederationConfig;
+use activitypub_federation::config::{FederationConfig, FederationMiddleware};
 use axum::{Router, routing::get};
 use tower_http::trace::TraceLayer;
 
@@ -8,7 +8,8 @@ pub mod routes;
 
 pub fn router(state: FederationConfig<AppHandle>) -> Router {
     Router::new()
+        .merge(routes::users::users_router())
         .route("/", get(health_check))
         .layer(TraceLayer::new_for_http())
-        .with_state(state)
+        .layer(FederationMiddleware::new(state))
 }