diff --git a/src/Plugin/Derivative/TextLongFieldTocBlockDeriver.php b/src/Plugin/Derivative/TextLongFieldTocBlockDeriver.php
index 359543f..3fe1dfd 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, private \Drupal\table_of_contents\TocTextFieldHelper $tocTextFieldHelper) {
     $this->entityTypeRepository = $entity_type_repository;
     $this->entityFieldManager = $entity_field_manager;
     $this->fieldTypeManager = $field_type_manager;
@@ -111,7 +111,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->tocTextFieldHelper,
       $container->get('context.repository')
     );
   }