diff --git a/islandora.tokens.inc b/islandora.tokens.inc
index fae9156b..fb7f2844 100644
--- a/islandora.tokens.inc
+++ b/islandora.tokens.inc
@@ -8,6 +8,8 @@
  */
 
 use Drupal\Core\Render\BubbleableMetadata;
+use Drupal\media\Entity\Media;
+use Drupal\file\Entity\File;
 
 /**
  * Implements hook_token_info().