1 files changed, 5 insertions, 6 deletions
diff --git a/src/web3_wrapper.ts b/src/web3_wrapper.ts
index 05a8dc063..5bff25233 100644
--- a/src/web3_wrapper.ts
+++ b/src/web3_wrapper.ts
@@ -2,16 +2,15 @@ import * as _ from 'lodash';
 import * as Web3 from 'web3';
 import * as BigNumber from 'bignumber.js';
 import promisify = require('es6-promisify');
-import {ZeroExError} from './types';
-import {assert} from './utils/assert';
+import {Web3Provider} from './types';
 
 export class Web3Wrapper {
     private web3: Web3;
-    constructor(web3: Web3) {
+    constructor(provider: Web3Provider) {
         this.web3 = new Web3();
-        this.web3.setProvider(web3.currentProvider);
+        this.web3.setProvider(provider);
     }
-    public setProvider(provider: Web3.Provider) {
+    public setProvider(provider: Web3Provider) {
         this.web3.setProvider(provider);
     }
     public isAddress(address: string): boolean {
@@ -25,7 +24,7 @@ export class Web3Wrapper {
         const nodeVersion = await promisify(this.web3.version.getNode)();
         return nodeVersion;
     }
-    public getCurrentProvider(): Web3.Provider {
+    public getCurrentProvider(): Web3Provider {
         return this.web3.currentProvider;
     }
     public async getNetworkIdIfExistsAsync(): Promise<number|undefined> {