diff --git a/contracts/TestToken.sol b/contracts/TestToken.sol
index a2d1725..b36fc78 100644
--- a/contracts/TestToken.sol
+++ b/contracts/TestToken.sol
@@ -6,9 +6,7 @@ import "@openzeppelin/contracts/token/ERC20/ERC20.sol";
 contract TestToken is ERC20 {
   constructor() ERC20("TestToken", "TST") {}
 
-  function mint(address[] memory holders, uint amount) public {
-    for (uint i=0; i<holders.length; i++) {
-      _mint(holders[i], amount);
-    }
+  function mint(address holder, uint amount) public {
+    _mint(holder, amount);
   }
 }
diff --git a/test/Stakes.test.js b/test/Stakes.test.js
index 70be394..6306603 100644
--- a/test/Stakes.test.js
+++ b/test/Stakes.test.js
@@ -13,7 +13,7 @@ describe("Stakes", function () {
     const TestToken = await ethers.getContractFactory("TestToken")
     token = await TestToken.deploy()
     stakes = await Stakes.deploy(token.address)
-    await token.mint([host.address], 1000)
+    await token.mint(host.address, 1000)
   })
 
   it("has zero stakes initially", async function () {
diff --git a/test/Storage.test.js b/test/Storage.test.js
index e3a62b1..b822e00 100644
--- a/test/Storage.test.js
+++ b/test/Storage.test.js
@@ -19,7 +19,8 @@ describe("Storage", function () {
     await deployments.fixture(['TestToken', 'Storage'])
     token = await ethers.getContract('TestToken')
     storage = await ethers.getContract('Storage')
-    await token.mint([client.address, host.address], 1000)
+    await token.mint(client.address, 1000)
+    await token.mint(host.address, 1000)
     stakeAmount = await storage.stakeAmount()
     slashMisses = await storage.slashMisses()
     slashPercentage = await storage.slashPercentage()