diff --git a/src/Form/AchievementEntityDeleteForm.php b/src/Form/AchievementEntityDeleteForm.php
index a9864bb..311274d 100755
--- a/src/Form/AchievementEntityDeleteForm.php
+++ b/src/Form/AchievementEntityDeleteForm.php
@@ -38,7 +38,7 @@ class AchievementEntityDeleteForm extends EntityConfirmFormBase {
   public function submitForm(array &$form, FormStateInterface $form_state) {
     $this->entity->delete();
 
-    drupal_set_message(
+    \Drupal::messenger()->addMessage(
       $this->t('content @type: deleted @label.',
         [
           '@type' => $this->entity->bundle(),
diff --git a/src/Form/AchievementEntityForm.php b/src/Form/AchievementEntityForm.php
index 94a2469..e6a8db3 100755
--- a/src/Form/AchievementEntityForm.php
+++ b/src/Form/AchievementEntityForm.php
@@ -251,13 +251,13 @@ class AchievementEntityForm extends EntityForm {
 
     switch ($status) {
       case SAVED_NEW:
-        drupal_set_message($this->t('Created the %label Achievement entity.', [
+        \Drupal::messenger()->addMessage($this->t('Created the %label Achievement entity.', [
           '%label' => $achievement_entity->label(),
         ]));
         break;
 
       default:
-        drupal_set_message($this->t('Saved the %label Achievement entity.', [
+        \Drupal::messenger()->addMessage($this->t('Saved the %label Achievement entity.', [
           '%label' => $achievement_entity->label(),
         ]));
     }