From 386340428959862f00f6ae066fdd4bbe50efeef5 Mon Sep 17 00:00:00 2001
From: Adam Gall <galladamj@gmail.com>
Date: Tue, 7 May 2024 13:53:17 -0400
Subject: [PATCH] Don't need to create a contract object, address will do

---
 src/models/AzoriusTxBuilder.ts | 13 ++++---------
 1 file changed, 4 insertions(+), 9 deletions(-)

diff --git a/src/models/AzoriusTxBuilder.ts b/src/models/AzoriusTxBuilder.ts
index 7fad01c0e3..5f82d93012 100644
--- a/src/models/AzoriusTxBuilder.ts
+++ b/src/models/AzoriusTxBuilder.ts
@@ -18,7 +18,6 @@ import {
   isAddress,
   isHex,
   encodeFunctionData,
-  GetContractReturnType,
   PublicClient,
   getContract,
 } from 'viem';
@@ -56,7 +55,7 @@ export class AzoriusTxBuilder extends BaseTxBuilder {
   public azoriusContract: Azorius | undefined;
   public linearVotingContract: LinearERC20Voting | undefined;
   public linearERC721VotingContract: LinearERC721Voting | undefined;
-  public votesTokenContract: GetContractReturnType<typeof VotesERC20Abi> | undefined;
+  public votesTokenContractAddress: Address | undefined;
 
   private votesERC20WrapperMasterCopyAddress: string;
   private votesERC20MasterCopyAddress: string;
@@ -248,14 +247,14 @@ export class AzoriusTxBuilder extends BaseTxBuilder {
   }
 
   public buildApproveClaimAllocation() {
-    if (!this.votesTokenContract) {
+    if (!this.votesTokenContractAddress) {
       return;
     }
 
     const azoriusGovernanceDaoData = this.daoData as AzoriusERC20DAO;
     return buildContractCallViem(
       VotesERC20Abi,
-      this.votesTokenContract.address,
+      this.votesTokenContractAddress,
       'approve',
       [this.predictedTokenClaimAddress, azoriusGovernanceDaoData.parentAllocationAmount],
       0,
@@ -569,11 +568,7 @@ export class AzoriusTxBuilder extends BaseTxBuilder {
         this.predictedStrategyAddress!,
         this.signerOrProvider,
       );
-      this.votesTokenContract = getContract({
-        abi: VotesERC20Abi,
-        address: this.predictedTokenAddress,
-        client: this.publicClient,
-      });
+      this.votesTokenContractAddress = this.predictedTokenAddress;
     } else if (daoData.votingStrategyType === VotingStrategyType.LINEAR_ERC721) {
       this.linearERC721VotingContract = LinearERC721Voting__factory.connect(
         this.predictedStrategyAddress!,