diff --git a/packages/token-distribution/contracts/GraphTokenLockWallet.sol b/packages/token-distribution/contracts/GraphTokenLockWallet.sol
index 3c240396e..724031edf 100644
--- a/packages/token-distribution/contracts/GraphTokenLockWallet.sol
+++ b/packages/token-distribution/contracts/GraphTokenLockWallet.sol
@@ -9,7 +9,7 @@ pragma experimental ABIEncoderV2;
 import { Address } from "@openzeppelin/contracts/utils/Address.sol";
 import { SafeMath } from "@openzeppelin/contracts/math/SafeMath.sol";
 
-import { GraphTokenLock, MathUtils } from "./GraphTokenLock.sol";
+import { GraphTokenLock } from "./GraphTokenLock.sol";
 import { IGraphTokenLock } from "./IGraphTokenLock.sol";
 import { IGraphTokenLockManager } from "./IGraphTokenLockManager.sol";
 
diff --git a/packages/token-distribution/test/tokenLockWallet.test.ts b/packages/token-distribution/test/tokenLockWallet.test.ts
index 362aabfe5..00146a03f 100644
--- a/packages/token-distribution/test/tokenLockWallet.test.ts
+++ b/packages/token-distribution/test/tokenLockWallet.test.ts
@@ -9,16 +9,12 @@ import { DeployOptions } from 'hardhat-deploy/types'
 
 import { GraphTokenLockManager, GraphTokenLockWallet, GraphTokenMock, StakingMock } from '../types'
 import { defaultInitArgs, Revocability, TokenLockParameters } from './config'
-import { Account, advanceBlocks, advanceTimeAndBlock, getAccounts, getContract, randomHexBytes, toGRT } from './network'
+import { Account, advanceTimeAndBlock, getAccounts, getContract, randomHexBytes, toGRT } from './network'
 
 const { AddressZero, MaxUint256 } = constants
 
 // -- Time utils --
 
-const advancePeriods = async (tokenLock: GraphTokenLockWallet, n = 1) => {
-  const periodDuration = await tokenLock.periodDuration()
-  return advanceTimeAndBlock(periodDuration.mul(n).toNumber()) // advance N period
-}
 const advanceToStart = async (tokenLock: GraphTokenLockWallet) => moveToTime(tokenLock, await tokenLock.startTime(), 60)
 const moveToTime = async (tokenLock: GraphTokenLockWallet, target: BigNumber, buffer: number) => {
   const ts = await tokenLock.currentTime()