diff --git a/src/Form/FirebasePhpSendTestMessageForm.php b/src/Form/FirebasePhpSendTestMessageForm.php
index d2367446ff8b36ecefadb19dd9990a491ff2ac32..9a40e6a40148fb99a052c59e7bbc4b5a8cbe8fbe 100644
--- a/src/Form/FirebasePhpSendTestMessageForm.php
+++ b/src/Form/FirebasePhpSendTestMessageForm.php
@@ -63,6 +63,11 @@ final class FirebasePhpSendTestMessageForm extends FormBase {
       '#maxlength' => 255,
       '#required' => TRUE,
     ];
+    $form['badge_count'] = [
+      '#type' => 'number',
+      '#title' => $this->t('Badge count'),
+      '#required' => FALSE,
+    ];
 
     // Group submit handlers in an actions element with a key of "actions" so
     // that it gets styled correctly, and so that other modules may add actions
@@ -89,6 +94,10 @@ final class FirebasePhpSendTestMessageForm extends FormBase {
         $form_state->getValue('title'),
         $form_state->getValue('message'),
         $form_state->getValue('device_token'),
+        NULL,
+        NULL,
+        NULL,
+        intval($form_state->getValue('badge_count')),
       );
       $this->messenger()->addStatus($this->t('The test message has been sent.'));
     }