diff --git a/foundry/src/TychoRouter.sol b/foundry/src/TychoRouter.sol
index 2b98667..dbdf089 100644
--- a/foundry/src/TychoRouter.sol
+++ b/foundry/src/TychoRouter.sol
@@ -369,7 +369,7 @@ contract TychoRouter is
         int256 amount0Delta,
         int256 amount1Delta,
         bytes calldata data
-    ) internal view returns (uint256 amountOwed, address tokenIn) {
+    ) internal view returns (uint256 amountIn, address tokenIn) {
         tokenIn = address(bytes20(data[0:20]));
         address tokenOut = address(bytes20(data[20:40]));
         uint24 poolFee = uint24(bytes3(data[40:43]));
@@ -379,9 +379,9 @@ contract TychoRouter is
             _usv3Factory, tokenIn, tokenOut, poolFee
         );
 
-        amountOwed =
+        amountIn =
             amount0Delta > 0 ? uint256(amount0Delta) : uint256(amount1Delta);
 
-        return (amountOwed, tokenIn);
+        return (amountIn, tokenIn);
     }
 }
diff --git a/src/encoding/models.rs b/src/encoding/models.rs
index 5e37500..4e2cbbd 100644
--- a/src/encoding/models.rs
+++ b/src/encoding/models.rs
@@ -65,7 +65,6 @@ pub struct Transaction {
 #[allow(dead_code)]
 pub struct EncodingContext {
     pub receiver: Bytes,
-    // TODO should we keep this?
     pub exact_out: bool,
     pub router_address: Bytes,
 }