Use an enum for contract name

1 files changed, 13 insertions, 12 deletions

diff --git a/packages/contracts/util/types.ts b/packages/contracts/util/types.ts
index baeb15577..0db04cd76 100644
--- a/packages/contracts/util/types.ts
+++ b/packages/contracts/util/types.ts
@@ -119,20 +119,21 @@ export enum ExchangeContractErrs {
     ERROR_INSUFFICIENT_BALANCE_OR_ALLOWANCE,
 }
 
-export type ArtifactContractName =
-    | 'TokenTransferProxy'
-    | 'TokenRegistry'
-    | 'MultiSigWalletWithTimeLock'
-    | 'Exchange'
-    | 'ZRXToken'
-    | 'DummyToken'
-    | 'DummyToken_v2'
-    | 'WETH9'
-    | 'MultiSigWalletWithTimeLockExceptRemoveAuthorizedAddress'
-    | 'MaliciousToken';
+export enum ContractName {
+    TokenTransferProxy = 'TokenTransferProxy',
+    TokenRegistry = 'TokenRegistry',
+    MultiSigWalletWithTimeLock = 'MultiSigWalletWithTimeLock',
+    Exchange = 'Exchange',
+    ZRXToken = 'ZRXToken',
+    DummyToken = 'DummyToken',
+    DummyToken_v2 = 'DummyToken_v2',
+    EtherToken = 'WETH9',
+    MultiSigWalletWithTimeLockExceptRemoveAuthorizedAddress = 'MultiSigWalletWithTimeLockExceptRemoveAuthorizedAddress',
+    MaliciousToken = 'MaliciousToken',
+}
 
 export interface Artifact {
-    contract_name: ArtifactContractName;
+    contract_name: ContractName;
     networks: {
         [networkId: number]: {
             abi: Web3.ContractAbi;