diff --git a/types/index.d.ts b/types/index.d.ts
index 5be3f8228..31040e3c1 100755
--- a/types/index.d.ts
+++ b/types/index.d.ts
@@ -70,10 +70,10 @@ export interface FirebasePlugin {
         error: (err: string) => void
     ): void
     unregister(): void
-    setBadgeNumber(
+    setBadgeNumber?(
         badgeNumber: number
     ): void
-    getBadgeNumber(
+    getBadgeNumber?(
         success: (badgeNumber: number) => void,
         error: (err: string) => void
     ): void