diff --git a/web/modules/custom/yufu_concept/src/Plugin/Block/ModerateFromDiffBlock.php b/web/modules/custom/yufu_concept/src/Plugin/Block/ModerateFromDiffBlock.php
index e61af86..06a9acd 100644
--- a/web/modules/custom/yufu_concept/src/Plugin/Block/ModerateFromDiffBlock.php
+++ b/web/modules/custom/yufu_concept/src/Plugin/Block/ModerateFromDiffBlock.php
@@ -4,7 +4,6 @@ declare(strict_types=1);
 
 namespace Drupal\yufu_concept\Plugin\Block;
 
-use Drupal\Core\Block\BlockBase;
 use Drupal\content_moderation_info_block\Plugin\Block\ContentModerationInfoBlock;
 use Drupal\content_moderation_info_block\Form\ContentModerationInfoBlockModerationForm;
 use Drupal\content_moderation_info_block\Form\ContentModerationInfoBlockPublishForm;
@@ -20,13 +19,13 @@ use Drupal\content_moderation_info_block\Form\ContentModerationInfoBlockPublishF
  * )
  */
 final class ModerateFromDiffBlock extends ContentModerationInfoBlock {
+
   /**
    * {@inheritdoc}
    */
   public function build() {
     $build = [];
     $configuration = $this->getConfiguration();
-    $entity = $this->getContextValue('entity');
 
     $routeMatcher = \Drupal::service('current_route_match');