From a056919507a9468e7bd7d54b5d7fa1c964abe89b Mon Sep 17 00:00:00 2001 From: Luke Van Seters Date: Thu, 20 Jan 2022 19:03:00 -0500 Subject: [PATCH] Fix test swaps --- tests/test_swaps.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_swaps.py b/tests/test_swaps.py index d5d8a1b..b1fd658 100644 --- a/tests/test_swaps.py +++ b/tests/test_swaps.py @@ -72,7 +72,7 @@ def test_swaps( from_address=alice_address, contract_address=first_pool_address, abi_name=UNISWAP_V2_PAIR_ABI_NAME, - protocol=None, + protocol=Protocol.uniswap_v2, function_signature="swap(uint256,uint256,address,bytes)", recipient_address=bob_address, recipient_input_key="to", @@ -93,7 +93,7 @@ def test_swaps( from_address=bob_address, contract_address=second_pool_address, abi_name=UNISWAP_V3_POOL_ABI_NAME, - protocol=None, + protocol=Protocol.uniswap_v2, function_signature="swap(address,bool,int256,uint160,bytes)", recipient_address=carl_address, recipient_input_key="recipient",