diff --git a/src/Plugin/Derivative/TextLongFieldTocBlockDeriver.php b/src/Plugin/Derivative/TextLongFieldTocBlockDeriver.php
index 359543f..760907f 100644
--- a/src/Plugin/Derivative/TextLongFieldTocBlockDeriver.php
+++ b/src/Plugin/Derivative/TextLongFieldTocBlockDeriver.php
@@ -91,7 +91,7 @@ final class TextLongFieldTocBlockDeriver extends DeriverBase implements Containe
    * @param Drupal\Core\Plugin\Context\ContextRepositoryInterface $context_repository
    *   The context repository.
    */
-  public function __construct(EntityTypeRepositoryInterface $entity_type_repository, EntityFieldManagerInterface $entity_field_manager, FieldTypePluginManagerInterface $field_type_manager, FormatterPluginManager $formatter_manager, EntityTypeBundleInfoInterface $bundle_info, TocTextFieldHelper $toc_field_helper, ContextRepositoryInterface $context_repository) {
+  public function __construct(EntityTypeRepositoryInterface $entity_type_repository, EntityFieldManagerInterface $entity_field_manager, FieldTypePluginManagerInterface $field_type_manager, FormatterPluginManager $formatter_manager, EntityTypeBundleInfoInterface $bundle_info, TocTextFieldHelper $toc_field_helper, ContextRepositoryInterface $context_repository, TocTextFieldHelper $tocFieldHelper) {
     $this->entityTypeRepository = $entity_type_repository;
     $this->entityFieldManager = $entity_field_manager;
     $this->fieldTypeManager = $field_type_manager;
@@ -99,6 +99,7 @@ final class TextLongFieldTocBlockDeriver extends DeriverBase implements Containe
     $this->formatterManager = $formatter_manager;
     $this->tocFieldHelper = $toc_field_helper;
     $this->contextRepository = $context_repository;
+    $this->tocFieldHelper = $tocFieldHelper;
   }
 
   /**
@@ -111,7 +112,7 @@ final class TextLongFieldTocBlockDeriver extends DeriverBase implements Containe
       $container->get('plugin.manager.field.field_type'),
       $container->get('plugin.manager.field.formatter'),
       $container->get('entity_type.bundle.info'),
-      $container->get(TocTextFieldHelper::class),
+      $this->tocFieldHelper,
       $container->get('context.repository')
     );
   }