diff --git a/index.js b/index.js
index b559434acb29f6d87e40112b468edf3a0db920d1..593992abcbe6b0fcfdab60b324aab9df850fe084 100644
--- a/index.js
+++ b/index.js
@@ -42,8 +42,8 @@ FCM.cancelAllLocalNotifications = () => {
     RNFIRMessaging.cancelAllLocalNotifications();
 };
 
-FCM.setBadgeNumber = () => {
-    RNFIRMessaging.setBadgeNumber();
+FCM.setBadgeNumber = (number) => {
+    RNFIRMessaging.setBadgeNumber(number);
 }
 
 FCM.getBadgeNumber = () => {