diff --git a/protocol-testing/src/encoding.rs b/protocol-testing/src/encoding.rs
index 7541bc7..936eb84 100644
--- a/protocol-testing/src/encoding.rs
+++ b/protocol-testing/src/encoding.rs
@@ -117,13 +117,10 @@ pub fn encode_swap(
         .wrap_err("Failed to encode solution")?[0]
         .clone();
 
-    let transaction = encode_tycho_router_call(
-        encoded_solution,
-        &solution,
-        &chain.wrapped_native_token().address,
-    )
-    .into_diagnostic()
-    .wrap_err("Failed to encode router calldata")?;
+    let transaction =
+        encode_tycho_router_call(encoded_solution, &solution, &chain.native_token().address)
+            .into_diagnostic()
+            .wrap_err("Failed to encode router calldata")?;
     Ok((transaction, solution))
 }
 
diff --git a/protocol-testing/src/main.rs b/protocol-testing/src/main.rs
index c01625a..1ae8748 100644
--- a/protocol-testing/src/main.rs
+++ b/protocol-testing/src/main.rs
@@ -81,7 +81,7 @@ impl Args {
 fn main() -> miette::Result<()> {
     // Load .env file before setting up logging
     dotenv().ok();
-    
+
     tracing_subscriber::fmt()
         .with_env_filter(EnvFilter::from_default_env())
         .with_target(false)