diff --git a/contracts/mock/tokens/StETHMock.sol b/contracts/mock/tokens/StETHMock.sol
index 4073bd7..c1092a7 100644
--- a/contracts/mock/tokens/StETHMock.sol
+++ b/contracts/mock/tokens/StETHMock.sol
@@ -11,7 +11,7 @@ contract StETHMock is ERC20, Ownable {
     constructor() ERC20("Staked Ether Mock", "stETHMock") {}
 
     function mint(address account_, uint256 amount_) external {
-        require(amount_ <= 10 * (10 ** decimals()), "StETHMock: amount is too big");
+        require(amount_ <= 1000 * (10 ** decimals()), "StETHMock: amount is too big");
 
         _mint(account_, amount_);
     }
diff --git a/test/Distribution.test.ts b/test/Distribution.test.ts
index 1807624..fcc0483 100644
--- a/test/Distribution.test.ts
+++ b/test/Distribution.test.ts
@@ -1312,7 +1312,7 @@ describe('Distribution', () => {
       expect(await depositToken.balanceOf(distribution)).to.eq(wei(20));
 
       await setNextTime(oneDay + oneDay);
-      await depositToken.setTotalPooledEther(wei(0.8, 25));
+      await depositToken.setBalanceMultiplicator(wei(0.8, 25));
       expect(await depositToken.balanceOf(distribution)).to.eq(wei(16));
 
       let tx = await distribution.withdraw(poolId, wei(999));
@@ -1330,7 +1330,7 @@ describe('Distribution', () => {
     it('should revert if trying to withdraw zero', async () => {
       await distribution.stake(poolId, wei(10));
 
-      await depositToken.setTotalPooledEther(wei(0.0001, 25));
+      await depositToken.setBalanceMultiplicator(wei(0.0001, 25));
 
       await distribution.withdraw(poolId, wei(10));
 
@@ -1582,7 +1582,7 @@ describe('Distribution', () => {
 
       await setTime(oneDay * 9999);
 
-      await depositToken.setTotalPooledEther(wei(0.5));
+      await depositToken.setBalanceMultiplicator(wei(0.5));
 
       const overplus = await distribution.overplus();
       expect(overplus).to.eq(0);
@@ -1590,7 +1590,7 @@ describe('Distribution', () => {
     it('should return overplus if deposited token increased', async () => {
       await distribution.stake(0, wei(1));
 
-      await depositToken.setTotalPooledEther(wei(2, 25));
+      await depositToken.setBalanceMultiplicator(wei(2, 25));
 
       let overplus = await distribution.overplus();
       expect(overplus).to.eq(wei(1));
@@ -1600,12 +1600,12 @@ describe('Distribution', () => {
       overplus = await distribution.overplus();
       expect(overplus).to.eq(wei(1));
 
-      await depositToken.setTotalPooledEther(wei(1, 25));
+      await depositToken.setBalanceMultiplicator(wei(1, 25));
 
       overplus = await distribution.overplus();
       expect(overplus).to.eq(0);
 
-      await depositToken.setTotalPooledEther(wei(5, 25));
+      await depositToken.setBalanceMultiplicator(wei(5, 25));
 
       overplus = await distribution.overplus();
       expect(overplus).to.eq(wei(5.5));
@@ -1626,7 +1626,7 @@ describe('Distribution', () => {
 
       await distribution.stake(1, wei(1));
 
-      await depositToken.setTotalPooledEther(wei(2, 25));
+      await depositToken.setBalanceMultiplicator(wei(2, 25));
 
       const overplus = await distribution.overplus();
       expect(overplus).to.eq(wei(1));