diff --git a/test_deployment/test_deploy.py b/test_deployment/test_deploy.py
index 9cd30e1..90a6831 100644
--- a/test_deployment/test_deploy.py
+++ b/test_deployment/test_deploy.py
@@ -19,7 +19,7 @@ def test_add_liquidity(
         self, admin, deploy, admin_mint_tokens, approve_admin, tokens_to_add
     ):
         pool = deploy[0]
-        tx = pool.add_liquidity(tokens_to_add, 0, False, sender=admin)
+        tx = pool.add_liquidity(tokens_to_add, 0, sender=admin)
         d_tokens = tx.return_value
         assert pool.balanceOf(admin) == pool.totalSupply() == d_tokens
 
@@ -35,14 +35,14 @@ def test_exchange(
         j,
     ):
         pool = deploy[0]
-        pool.add_liquidity(tokens_to_add, 0, False, sender=admin)
+        pool.add_liquidity(tokens_to_add, 0, sender=admin)
 
         if i == 0:
             amount = 100 * 10 * 6
         else:
             amount = 1 * 10**6
 
-        tx = pool.exchange_underlying(
+        tx = pool.exchange_received(
             i,
             j,
             amount,
@@ -59,7 +59,7 @@ def test_remove_liquidity_one(
         self, admin, deploy, admin_mint_tokens, approve_admin, tokens_to_add, i
     ):
         pool = deploy[0]
-        pool.add_liquidity(tokens_to_add, 0, False, sender=admin)
+        pool.add_liquidity(tokens_to_add, 0, sender=admin)
 
         bal = pool.balanceOf(admin)
         coin_contract = deploy[1][i]
@@ -69,7 +69,6 @@ def test_remove_liquidity_one(
             bal // 2,
             i,
             0,
-            False,
             sender=admin,
         )  # noqa: E501