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