diff --git a/src/index.js b/src/index.js
index 835a59432..28570ec83 100644
--- a/src/index.js
+++ b/src/index.js
@@ -1,6 +1,6 @@
-import { NativeModules, Platform } from "react-native";
+import { Platform } from "react-native";
 
-const { RNBranch } = NativeModules;
+import RNBranch from './RNBranch';
 
 import createBranchUniversalObject from "./branchUniversalObject";
 import BranchEvent from "./BranchEvent";
diff --git a/test/BranchEvent.test.js b/test/BranchEvent.test.js
index a23a768c2..d4dd6d98c 100644
--- a/test/BranchEvent.test.js
+++ b/test/BranchEvent.test.js
@@ -1,5 +1,4 @@
-import { NativeModules } from 'react-native'
-const { RNBranch } = NativeModules
+import RNBranch from '../src/RNBranch'
 import BranchEvent  from '../src/BranchEvent'
 
 // --- Constant mapping ---
diff --git a/test/BranchSubscriber.test.js b/test/BranchSubscriber.test.js
index 760313072..f8916446a 100644
--- a/test/BranchSubscriber.test.js
+++ b/test/BranchSubscriber.test.js
@@ -1,5 +1,4 @@
-import { NativeModules } from 'react-native'
-const { RNBranch } = NativeModules
+import RNBranch from '../src/RNBranch'
 import BranchSubscriber from '../src/BranchSubscriber'
 
 test('default initializes with no options', () => {