From 3cd03ed0f15039da617000681ae07658b088e6aa Mon Sep 17 00:00:00 2001
From: Lawrence Forman <lawrence@0xproject.com>
Date: Thu, 20 Aug 2020 16:55:47 -0400
Subject: [PATCH] `@0x/migrations`: Update EP migration

---
 packages/migrations/CHANGELOG.json   |  4 ++++
 packages/migrations/src/migration.ts | 18 ++++--------------
 2 files changed, 8 insertions(+), 14 deletions(-)

diff --git a/packages/migrations/CHANGELOG.json b/packages/migrations/CHANGELOG.json
index e2b39f9841..12fd9c63e2 100644
--- a/packages/migrations/CHANGELOG.json
+++ b/packages/migrations/CHANGELOG.json
@@ -9,6 +9,10 @@
             {
                 "note": "Refactor `migration.ts` a little",
                 "pr": 2656
+            },
+            {
+                "note": "Update EP migration.",
+                "pr": "TODO"
             }
         ]
     },
diff --git a/packages/migrations/src/migration.ts b/packages/migrations/src/migration.ts
index 67af1f6f78..df0bf4c333 100644
--- a/packages/migrations/src/migration.ts
+++ b/packages/migrations/src/migration.ts
@@ -31,11 +31,9 @@ import {
     artifacts as exchangeProxyArtifacts,
     FillQuoteTransformerContract,
     fullMigrateAsync as fullMigrateExchangeProxyAsync,
-    ITokenSpenderContract,
-    ITransformERC20Contract,
+    IZeroExContract,
     PayTakerTransformerContract,
     WethTransformerContract,
-    ZeroExContract,
 } from '@0x/contracts-zero-ex';
 import { Web3ProviderEngine, ZeroExProvider } from '@0x/subproviders';
 import { BigNumber, providerUtils } from '@0x/utils';
@@ -400,7 +398,7 @@ async function _migrateExchangeProxyAsync(
     txDefaults: TxData,
 ): Promise<
     [
-        ZeroExContract,
+        IZeroExContract,
         FillQuoteTransformerContract,
         PayTakerTransformerContract,
         WethTransformerContract,
@@ -410,16 +408,8 @@ async function _migrateExchangeProxyAsync(
     ]
 > {
     const exchangeProxy = await fullMigrateExchangeProxyAsync(txDefaults.from, provider, txDefaults);
-    const exchangeProxyAllowanceTargetAddress = await new ITokenSpenderContract(
-        exchangeProxy.address,
-        provider,
-        txDefaults,
-    )
-        .getAllowanceTarget()
-        .callAsync();
-    const exchangeProxyFlashWalletAddress = await new ITransformERC20Contract(exchangeProxy.address, provider)
-        .getTransformWallet()
-        .callAsync();
+    const exchangeProxyAllowanceTargetAddress = await exchangeProxy.getAllowanceTarget().callAsync();
+    const exchangeProxyFlashWalletAddress = await exchangeProxy.getTransformWallet().callAsync();
 
     // Deploy transformers.
     const fillQuoteTransformer = await FillQuoteTransformerContract.deployFrom0xArtifactAsync(