From f1e275c1e46aafbedf3e3a2bd87f52effdd45651 Mon Sep 17 00:00:00 2001
From: dtfiedler <dylan@ar.io>
Date: Mon, 30 Oct 2023 16:38:41 -0600
Subject: [PATCH] chore(tests): fix addFunds call within extend tests

---
 tests/extend.test.ts | 5 ++++-
 tests/setup.jest.ts  | 1 -
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/tests/extend.test.ts b/tests/extend.test.ts
index 3229a727..9cf867af 100644
--- a/tests/extend.test.ts
+++ b/tests/extend.test.ts
@@ -38,7 +38,10 @@ describe('Extend', () => {
       );
       contract = warp.pst(srcContractId).connect(nonContractOwner);
       emptyWalletCaller = await arweave.wallets.generate();
-      await addFunds(arweave, emptyWalletCaller);
+      const emptyWalletAddress = await arweave.wallets.getAddress(
+        emptyWalletCaller,
+      );
+      await addFunds(arweave, emptyWalletAddress);
     });
 
     afterEach(() => {
diff --git a/tests/setup.jest.ts b/tests/setup.jest.ts
index 0209efd4..6976786b 100644
--- a/tests/setup.jest.ts
+++ b/tests/setup.jest.ts
@@ -2,7 +2,6 @@ import * as fs from 'fs';
 import path from 'path';
 
 import { WALLETS_TO_CREATE } from './utils/constants';
-// import { WALLETS_TO_CREATE } from './utils/constants';
 import { createLocalWallet, setupInitialContractState } from './utils/helper';
 import { arlocal, arweave, warp } from './utils/services';