diff --git a/src/Form/WebformDigestDeleteForm.php b/src/Form/WebformDigestDeleteForm.php
index b20c43f..12e1218 100644
--- a/src/Form/WebformDigestDeleteForm.php
+++ b/src/Form/WebformDigestDeleteForm.php
@@ -38,14 +38,12 @@ class WebformDigestDeleteForm extends EntityConfirmFormBase {
   public function submitForm(array &$form, FormStateInterface $form_state) {
     $this->entity->delete();
 
-    drupal_set_message(
-      $this->t('content @type: deleted @label.',
-        [
-          '@type' => $this->entity->bundle(),
-          '@label' => $this->entity->label(),
-        ]
-        )
-    );
+    $this->messenger()->addStatus($this->t('content @type: deleted @label.',
+      [
+        '@type' => $this->entity->bundle(),
+        '@label' => $this->entity->label(),
+      ]
+      ));
 
     $form_state->setRedirectUrl($this->getCancelUrl());
   }
diff --git a/src/Form/WebformDigestForm.php b/src/Form/WebformDigestForm.php
index e3e7906..b05a241 100644
--- a/src/Form/WebformDigestForm.php
+++ b/src/Form/WebformDigestForm.php
@@ -105,13 +105,13 @@ class WebformDigestForm extends EntityForm {
 
     switch ($status) {
       case SAVED_NEW:
-        drupal_set_message($this->t('Created the %label Webform digest.', [
+        $this->messenger()->addStatus($this->t('Created the %label Webform digest.', [
           '%label' => $webform_digest->label(),
         ]));
         break;
 
       default:
-        drupal_set_message($this->t('Saved the %label Webform digest.', [
+        $this->messenger()->addStatus($this->t('Saved the %label Webform digest.', [
           '%label' => $webform_digest->label(),
         ]));
     }
diff --git a/webform_digests.info.yml b/webform_digests.info.yml
index 449e2a6..4785a78 100644
--- a/webform_digests.info.yml
+++ b/webform_digests.info.yml
@@ -2,6 +2,7 @@ name: Webform Submissions Notification
 type: module
 description: Send a digest for webform submissions
 core: 8.x
+core_version_requirement: ^8 || ^9
 package: Webform
 dependencies:
   - webform:webform