Remove truffle from tests

1 files changed, 4 insertions, 3 deletions

diff --git a/packages/contracts/util/balances.ts b/packages/contracts/util/balances.ts
index 6a1659ab1..5bcb5145f 100644
--- a/packages/contracts/util/balances.ts
+++ b/packages/contracts/util/balances.ts
@@ -1,12 +1,13 @@
 import { BigNumber } from '@0xproject/utils';
 import * as _ from 'lodash';
+import * as Web3 from 'web3';
 
-import { BalancesByOwner, ContractInstance } from './types';
+import { BalancesByOwner } from './types';
 
 export class Balances {
-    private _tokenContractInstances: ContractInstance[];
+    private _tokenContractInstances: Web3.ContractInstance[];
     private _ownerAddresses: string[];
-    constructor(tokenContractInstances: ContractInstance[], ownerAddresses: string[]) {
+    constructor(tokenContractInstances: Web3.ContractInstance[], ownerAddresses: string[]) {
         this._tokenContractInstances = tokenContractInstances;
         this._ownerAddresses = ownerAddresses;
     }