diff --git a/core/modules/content_moderation/src/Plugin/Menu/EditTab.php b/core/modules/content_moderation/src/Plugin/Menu/EditTab.php
index 2980ae8..47a573b 100644
--- a/core/modules/content_moderation/src/Plugin/Menu/EditTab.php
+++ b/core/modules/content_moderation/src/Plugin/Menu/EditTab.php
@@ -2,6 +2,8 @@
 
 namespace Drupal\content_moderation\Plugin\Menu;
 
+use Drupal\Core\Entity\ContentEntityInterface;
+use Drupal\Core\Entity\EntityInterface;
 use Drupal\Core\Menu\LocalTaskDefault;
 use Drupal\Core\Plugin\ContainerFactoryPluginInterface;
 use Drupal\Core\Routing\RouteMatchInterface;
@@ -78,15 +80,19 @@ public function getRouteParameters(RouteMatchInterface $route_match) {
    * {@inheritdoc}
    */
   public function getTitle() {
-    if (!$this->moderationInfo->isModeratedEntity($this->entity)) {
-      // Moderation isn't enabled.
-      return parent::getTitle();
+    if ($this->entity instanceof EntityInterface) {
+      if (!$this->moderationInfo->isModeratedEntity($this->entity)) {
+        // Moderation isn't enabled.
+        return parent::getTitle();
+      }
     }
 
     // @todo https://www.drupal.org/node/2779933 write a test for this.
-    return $this->moderationInfo->isLiveRevision($this->entity)
-      ? $this->t('New draft')
-      : $this->t('Edit draft');
+    if ($this->entity instanceof ContentEntityInterface) {
+      return $this->moderationInfo->isLiveRevision($this->entity)
+        ? $this->t('New draft')
+        : $this->t('Edit draft');
+    }
   }
 
   /**