1 files changed, 4 insertions, 4 deletions
diff --git a/src/transport/firewall.rs b/src/transport/firewall.rs
index 917fe5f..b8e05cd 100644
--- a/src/transport/firewall.rs
+++ b/src/transport/firewall.rs
@@ -69,8 +69,8 @@ impl Transport for FirewallTransport {
         self.inner.local_id()
     }
 
-    async fn resolve_peer(&self, hint: &str) -> Result<String> {
-        self.inner.resolve_peer(hint).await
+    async fn resolve_seed(&self, seed: &str) -> Result<String> {
+        self.inner.resolve_seed(seed).await
     }
 
     fn encode(&self, peer: &str, message: &ParsedMessage) -> Result<EncodedMessage> {
@@ -89,12 +89,12 @@ impl Transport for FirewallTransport {
         let firewall = Arc::clone(&self.firewall);
 
         self.inner
-            .start_consumer(Box::new(move |sender, message| {
+            .start_consumer(Box::new(move |sender, message, payload_len| {
                 if firewall.blocks_inbound(&sender) {
                     tracing::trace!(peer = %sender, "Firewall dropped inbound message");
                     return Box::pin(async { Ok(()) });
                 }
-                handler(sender, message)
+                handler(sender, message, payload_len)
             }))
             .await
     }