diff --git a/src/TreeBuilder.php b/src/TreeBuilder.php
index 8e58ba3..e5d8799 100644
--- a/src/TreeBuilder.php
+++ b/src/TreeBuilder.php
@@ -6,9 +6,12 @@ use Drupal\Core\Cache\Cache;
 use Drupal\Core\Cache\CacheBackendInterface;
 use Drupal\Core\Language\LanguageManagerInterface;
 use Drupal\Core\Render\BubbleableMetadata;
+use Drupal\Core\StringTranslation\StringTranslationTrait;
 
 class TreeBuilder implements TreeBuilderInterface {
 
+  use StringTranslationTrait;
+
   /**
    * @var \Drupal\token\Token
    */
@@ -97,7 +100,7 @@ class TreeBuilder implements TreeBuilderInterface {
       '#show_nested' => $options['show_nested'],
       '#click_insert' => $options['click_insert'],
       '#columns' => ['name', 'token', 'description'],
-      '#empty' => t('No tokens available'),
+      '#empty' => $this->t('No tokens available'),
     ];
 
     return $element;
diff --git a/token.tokens.inc b/token.tokens.inc
index 7ee4291..763c712 100644
--- a/token.tokens.inc
+++ b/token.tokens.inc
@@ -25,6 +25,7 @@ use Symfony\Cmf\Component\Routing\RouteObjectInterface;
 use Drupal\Core\TypedData\PrimitiveInterface;
 use Drupal\Core\Field\FieldStorageDefinitionInterface;
 use Drupal\Core\Entity\ContentEntityTypeInterface;
+
 use Drupal\image\Entity\ImageStyle;
 
 /**