diff --git a/modules/taxonomy_diff_ui/taxonomy_diff_ui.module b/modules/taxonomy_diff_ui/taxonomy_diff_ui.module
index 9f49d9e..9961945 100644
--- a/modules/taxonomy_diff_ui/taxonomy_diff_ui.module
+++ b/modules/taxonomy_diff_ui/taxonomy_diff_ui.module
@@ -1,12 +1,13 @@
 <?php
 
+use Drupal\Core\Entity\EntityInterface;
 use Drupal\taxonomy\TermInterface;
 
 /**
  * Implements hook_entity_presave().
  * 
  */
-function taxonomy_diff_ui_entity_presave(Drupal\Core\Entity\EntityInterface $entity) {
+function taxonomy_diff_ui_entity_presave(EntityInterface $entity) {
   if ($entity instanceof TermInterface) {
     $uid = \Drupal::currentUser()->id();
     $entity->setRevisionUserId($uid);
@@ -25,4 +26,4 @@ function taxonomy_diff_ui_entity_type_alter(array &$entity_types) {
   
   // Set the revision link.
   $entity_types['taxonomy_term']->setLinkTemplate('revision', '/taxonomy/term/{taxonomy_term}/revisions/{taxonomy_term_revision}/view');
-}
\ No newline at end of file
+}