diff --git a/tests/.babelrc b/tests/.babelrc
index 853eb4ad..29569a9e 100644
--- a/tests/.babelrc
+++ b/tests/.babelrc
@@ -22,7 +22,7 @@
             "instrument": true,
             "relativePath": false,
             "include": [
-              "**/dist/**"
+              "**/react-native-firebase/src/**"
             ]
           }
         ]
diff --git a/tests/app.js b/tests/app.js
index 9339e62f..86cc0463 100755
--- a/tests/app.js
+++ b/tests/app.js
@@ -3,7 +3,7 @@ import React, { Component } from 'react';
 import { AppRegistry, Text, View } from 'react-native';
 
 import bridge from 'jet/platform/react-native';
-import firebase from 'react-native-firebase';
+import firebase from 'react-native-firebase/src';
 
 require('sinon');
 require('should-sinon');
diff --git a/tests/package.json b/tests/package.json
index 96429d51..c8e7ad83 100755
--- a/tests/package.json
+++ b/tests/package.json
@@ -57,13 +57,13 @@
     "functions": 95,
     "branches": 95,
     "include": [
-      "**/dist/**"
+      "**/react-native-firebase/src/**"
     ],
     "exclude": [
       "node_modules",
-      "**/dist/utils/emitter**",
-      "**/dist/modules/admob**",
-      "**/dist/modules/auth/phone**"
+      "**/react-native-firebase/src/utils/emitter**",
+      "**/react-native-firebase/src/modules/admob**",
+      "**/react-native-firebase/src/modules/auth/phone**"
     ],
     "sourceMap": true,
     "instrument": true,