From a5692690dc3ba394b328bb50689d687230bddb05 Mon Sep 17 00:00:00 2001
From: Amir Bandeali <abandeali1@gmail.com>
Date: Sun, 29 Sep 2019 15:45:58 -0700
Subject: [PATCH] Fix protocol fees tests

---
 contracts/staking/test/unit_tests/protocol_fees_test.ts | 3 ---
 contracts/staking/test/utils/api_wrapper.ts             | 1 -
 2 files changed, 4 deletions(-)

diff --git a/contracts/staking/test/unit_tests/protocol_fees_test.ts b/contracts/staking/test/unit_tests/protocol_fees_test.ts
index a89a0f4d74..de2f8bf29b 100644
--- a/contracts/staking/test/unit_tests/protocol_fees_test.ts
+++ b/contracts/staking/test/unit_tests/protocol_fees_test.ts
@@ -28,7 +28,6 @@ blockchainTests('Protocol Fees unit tests', env => {
     let exchangeAddress: string;
     let notExchangeAddress: string;
     let testContract: TestProtocolFeesContract;
-    let wethAssetData: string;
     let minimumStake: BigNumber;
 
     before(async () => {
@@ -46,7 +45,6 @@ blockchainTests('Protocol Fees unit tests', env => {
             exchangeAddress,
         );
 
-        wethAssetData = await testContract.getWethAssetData.callAsync();
         minimumStake = (await testContract.getParams.callAsync())[2];
     });
 
@@ -230,7 +228,6 @@ blockchainTests('Protocol Fees unit tests', env => {
                 );
                 expect(logsArgs.length).to.eq(1);
                 for (const args of logsArgs) {
-                    expect(args.assetData).to.eq(wethAssetData);
                     expect(args.from).to.eq(fromAddress);
                     expect(args.to).to.eq(testContract.address);
                     expect(args.amount).to.bignumber.eq(amount);
diff --git a/contracts/staking/test/utils/api_wrapper.ts b/contracts/staking/test/utils/api_wrapper.ts
index 6105715684..9c84859697 100644
--- a/contracts/staking/test/utils/api_wrapper.ts
+++ b/contracts/staking/test/utils/api_wrapper.ts
@@ -243,7 +243,6 @@ export async function deployAndConfigureContractsAsync(
         env.txDefaults,
         artifacts,
         wethContract.address,
-        erc20ProxyContract.address,
         zrxVaultContract.address,
     );