diff --git a/tests/android/build.gradle b/tests/android/build.gradle
index e176d9a5..59621ffe 100644
--- a/tests/android/build.gradle
+++ b/tests/android/build.gradle
@@ -5,7 +5,7 @@ buildscript {
         jcenter()
     }
     dependencies {
-        classpath 'com.android.tools.build:gradle:2.3.2'
+        classpath 'com.android.tools.build:gradle:2.3.3'
         classpath 'com.google.gms:google-services:3.1.0'
         classpath 'com.google.firebase:firebase-plugins:1.1.0'
     }
diff --git a/tests/android/gradle/wrapper/gradle-wrapper.properties b/tests/android/gradle/wrapper/gradle-wrapper.properties
index c17349ca..37253f59 100644
--- a/tests/android/gradle/wrapper/gradle-wrapper.properties
+++ b/tests/android/gradle/wrapper/gradle-wrapper.properties
@@ -4,5 +4,5 @@ distributionPath=wrapper/dists
 zipStoreBase=GRADLE_USER_HOME
 zipStorePath=wrapper/dists
 #distributionUrl=https\://services.gradle.org/distributions/gradle-3.3-all.zip
-distributionUrl=https\://services.gradle.org/distributions/gradle-4.0-milestone-1-all.zip
+distributionUrl=https\://services.gradle.org/distributions/gradle-4.1-all.zip
 
diff --git a/tests/index.android.js b/tests/index.android.js
index 9e8fc32c..de86556c 100644
--- a/tests/index.android.js
+++ b/tests/index.android.js
@@ -1,5 +1,5 @@
 import { AppRegistry } from 'react-native';
-import bootstrap from './src/main';
+import bootstrap from './src/bench';
 
 
 AppRegistry.registerComponent('ReactNativeFirebaseDemo', () => bootstrap);