diff --git a/message_digest_ui/message_digest_ui.install b/message_digest_ui/message_digest_ui.install
index e95972a..015a87d 100644
--- a/message_digest_ui/message_digest_ui.install
+++ b/message_digest_ui/message_digest_ui.install
@@ -13,7 +13,7 @@
  */
 function message_digest_ui_install() {
   // Add the `message_digest` field to the user form.
-  $display = entity_get_form_display('user', 'user', 'default');
+  $display = \Drupal::service('entity_display.repository')->getFormDisplay('user', 'user', 'default');
   if (!$display->getComponent('message_digest')) {
     $settings = [
       'type' => 'options_select',
@@ -26,7 +26,7 @@ function message_digest_ui_install() {
   // Add the `message_digest` field to flagging types.
   $prefix = \Drupal::config('message_subscribe_email.settings')->get('flag_prefix');
   foreach (['node', 'term', 'user'] as $name) {
-    $display = entity_get_form_display('flagging', $prefix . '_' . $name, 'default');
+    $display = \Drupal::service('entity_display.repository')->getFormDisplay('flagging', $prefix . '_' . $name, 'default');
     if (!$display->getComponent('message_digest')) {
       $settings = [
         'type' => 'options_select',