diff --git a/index.js b/index.js
index 112e8a3..b0845bd 100644
--- a/index.js
+++ b/index.js
@@ -7,8 +7,8 @@ const PreferredAndroidMarket = {
   OTHER: 'Other',
 }
 
-function appUpgradeVersionCheck(appInfo, xApiKey, alertInfo) {
-  versionCheck(appInfo, xApiKey, alertInfo);
+function appUpgradeVersionCheck(appInfo, xApiKey, alertInfo, baseUrl) {
+  versionCheck(appInfo, xApiKey, alertInfo, baseUrl);
 }
 
 export { appUpgradeVersionCheck, PreferredAndroidMarket };
diff --git a/src/version-check.js b/src/version-check.js
index f7464ce..d4976fb 100644
--- a/src/version-check.js
+++ b/src/version-check.js
@@ -3,11 +3,11 @@ import { checkVersionWithAppUpgrade } from "./api";
 import { Platform, Alert, Linking } from "react-native";
 import { PreferredAndroidMarket } from "app-upgrade-react-native-sdk";
 
-async function versionCheck(appInfo, xApiKey, alertInfo) {
+async function versionCheck(appInfo, xApiKey, alertInfo, baseUrl) {
   const isValid = validate(appInfo, xApiKey);
 
   if (isValid) {
-    const version = await checkVersionWithAppUpgrade(appInfo, xApiKey);
+    const version = await checkVersionWithAppUpgrade(appInfo, xApiKey, baseUrl);
     if (!version) {
       console.error("App Upgrade Error: Version is null.");
     } else if (version && version.found === true) {