1 files changed, 9 insertions, 10 deletions

diff --git a/packages/dev-utils/test/rpc_test.ts b/packages/dev-utils/test/rpc_test.ts
index 2f0e5ff7f..1bdea3613 100644
--- a/packages/dev-utils/test/rpc_test.ts
+++ b/packages/dev-utils/test/rpc_test.ts
@@ -3,18 +3,17 @@ import { Web3Wrapper } from '@0xproject/web3-wrapper';
 import * as chai from 'chai';
 import 'mocha';
 
-import { RPC, web3Factory } from '../src';
+import { web3Factory } from '../src';
 
 const expect = chai.expect;
 
 describe('RPC tests', () => {
-    const web3 = web3Factory.create();
-    const web3Wrapper = new Web3Wrapper(web3.currentProvider);
-    const rpc = new RPC();
+    const web3Provider = web3Factory.getRpcProvider();
+    const web3Wrapper = new Web3Wrapper(web3Provider);
     describe('#mineBlockAsync', () => {
         it('increases block number when called', async () => {
             const blockNumberBefore = await web3Wrapper.getBlockNumberAsync();
-            await rpc.mineBlockAsync();
+            await web3Wrapper.mineBlockAsync();
             const blockNumberAfter = await web3Wrapper.getBlockNumberAsync();
             expect(blockNumberAfter).to.be.equal(blockNumberBefore + 1);
         });
@@ -23,8 +22,8 @@ describe('RPC tests', () => {
         it('increases time when called', async () => {
             const TIME_DELTA = 1000;
             const blockTimestampBefore = await web3Wrapper.getBlockTimestampAsync(BlockParamLiteral.Latest);
-            await rpc.increaseTimeAsync(TIME_DELTA);
-            await rpc.mineBlockAsync();
+            await web3Wrapper.increaseTimeAsync(TIME_DELTA);
+            await web3Wrapper.mineBlockAsync();
             const blockTimestampAfter = await web3Wrapper.getBlockTimestampAsync(BlockParamLiteral.Latest);
             expect(blockTimestampAfter).to.be.at.least(blockTimestampBefore + TIME_DELTA);
         });
@@ -32,9 +31,9 @@ describe('RPC tests', () => {
     describe('#takeSnapshotAsync/revertSnapshotAsync', () => {
         it('reverts changes in between', async () => {
             const blockNumberBefore = await web3Wrapper.getBlockNumberAsync();
-            const snapshotId = await rpc.takeSnapshotAsync();
-            await rpc.mineBlockAsync();
-            await rpc.revertSnapshotAsync(snapshotId);
+            const snapshotId = await web3Wrapper.takeSnapshotAsync();
+            await web3Wrapper.mineBlockAsync();
+            await web3Wrapper.revertSnapshotAsync(snapshotId);
             const blockNumberAfter = await web3Wrapper.getBlockNumberAsync();
             expect(blockNumberAfter).to.be.equal(blockNumberBefore);
         });