Fix add tokens

Signed-off-by: deniallugo <[email protected]>
juliamist · Jun 1, 2021 · fabd8b2 · fabd8b2
1 parent 4afd2a9
commit fabd8b2
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/contracts/scripts/governance-add-erc20.ts b/contracts/scripts/governance-add-erc20.ts
@@ -18,7 +18,7 @@ async function governanceAddToken(address: string) {
     console.log('Adding new ERC20 token to network: ', address);
 
     const tx = await deployer
-        .governanceContract(governorWallet)
+        .tokenGovernanceContract(governorWallet)
         .addToken(address, { gasLimit: BigNumber.from('1000000') });
     console.log('tx hash: ', tx.hash);
     const receipt = await tx.wait();

diff --git a/contracts/src.ts/deploy.ts b/contracts/src.ts/deploy.ts
@@ -18,7 +18,9 @@ import {
     ZkSync,
     ZkSyncFactory,
     ForcedExit,
-    ForcedExitFactory
+    ForcedExitFactory,
+    TokenGovernanceFactory,
+    TokenGovernance
 } from '../typechain';
 
 export interface Contracts {
@@ -452,6 +454,10 @@ export class Deployer {
         return GovernanceFactory.connect(this.addresses.Governance, signerOrProvider);
     }
 
+    public tokenGovernanceContract(signerOrProvider: Signer | providers.Provider): TokenGovernance {
+        return TokenGovernanceFactory.connect(this.addresses.TokenGovernance, signerOrProvider);
+    }
+
     public zkSyncContract(signerOrProvider: Signer | providers.Provider): ZkSync {
         return ZkSyncFactory.connect(this.addresses.ZkSync, signerOrProvider);
     }