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';