Don't create whole web3 object in beta tests

2 files changed, 3 insertions, 9 deletions

diff --git a/test/utils/web3_factory.ts b/test/utils/web3_factory.ts
index 863fa1d07..4785fa858 100644
--- a/test/utils/web3_factory.ts
+++ b/test/utils/web3_factory.ts
@@ -17,12 +17,6 @@ export const web3Factory = {
         web3.setProvider(provider);
         return web3;
     },
-    createBeta(): any {
-        const provider_beta = this.getProviderBeta();
-        const web3_beta = new Web3_beta();
-        web3_beta.setProvider(provider_beta);
-        return web3_beta;
-    },
     getRpcProvider(hasAddresses: boolean = true): Web3.Provider {
         const provider = new ProviderEngine();
         const rpcUrl = `http://${constants.RPC_HOST}:${constants.RPC_PORT}`;
@@ -35,7 +29,7 @@ export const web3Factory = {
         provider.start();
         return provider;
     },
-    getProviderBeta(): any {
+    getProviderBeta(): Web3.Provider {
         const rpcUrl = `http://${constants.RPC_HOST}:${constants.RPC_PORT}`;
         const providerBeta = new Web3_beta.providers.HttpProvider(rpcUrl);
         return providerBeta;
diff --git a/test/web3_beta_test.ts b/test/web3_beta_test.ts
index 2fe962555..e34812243 100644
--- a/test/web3_beta_test.ts
+++ b/test/web3_beta_test.ts
@@ -8,8 +8,8 @@ chaiSetup.configure();
 const expect = chai.expect;
 
 describe('ZeroEx with beta web3', () => {
-    const web3_beta = web3Factory.createBeta();
-    const zeroEx = new ZeroEx(web3_beta.currentProvider);
+    const web3_beta_provider = web3Factory.getProviderBeta();
+    const zeroEx = new ZeroEx(web3_beta_provider);
     it('is able to make a call using a beta provider', async () => {
         await zeroEx.tokenRegistry.getTokenAddressesAsync();
     });