diff --git a/contrib/ui/entity_template_ui.module b/contrib/ui/entity_template_ui.module
index b200c3e..4377820 100644
--- a/contrib/ui/entity_template_ui.module
+++ b/contrib/ui/entity_template_ui.module
@@ -1,5 +1,10 @@
 <?php
 
+use Drupal\entity_template_ui\Entity\TemplateBuilderHtmlRouteProvider;
+use Drupal\entity_template_ui\Form\TemplateBuilderForm;
+use Drupal\entity_template_ui\Entity\TemplateBlueprintListBuilder;
+use Drupal\Core\Entity\Routing\DefaultHtmlRouteProvider;
+use Drupal\entity_template_ui\Form\TemplateBlueprintForm;
 use Drupal\entity_template\Plugin\EntityTemplate\Component\InlineTemplate;
 use Drupal\entity_template_ui\Plugin\EntityTemplate\Template\BlueprintTemplateUI;
 use Drupal\entity_template_ui\Plugin\EntityTemplate\Template\InlineTemplateUI;
@@ -15,7 +20,7 @@ function entity_template_ui_entity_type_build(array &$entity_types) {
   /** @var \Drupal\Core\Config\Entity\ConfigEntityType $entity_type */
   $entity_type = $entity_types['entity_template_builder'];
   $entity_type->setHandlerClass("route_provider", [
-    "html" => \Drupal\entity_template_ui\Entity\TemplateBuilderHtmlRouteProvider::class,
+    "html" => TemplateBuilderHtmlRouteProvider::class,
   ]);
   //$entity_type->setLinkTemplate("collection", "/admin/content/entity_template/builder");
   $entity_type->setLinkTemplate("canonical", "/admin/content/entity_template/builder/manage/{entity_template_builder}");
@@ -28,19 +33,19 @@ function entity_template_ui_entity_type_build(array &$entity_types) {
     "/admin/content/entity_template/builder/manage/{entity_template_builder}/edit_default_blueprint"
   );
   $entity_type->setLinkTemplate("add-form", "/admin/content/entity_template/builder/add");
-  $entity_type->setFormClass("default", \Drupal\entity_template_ui\Form\TemplateBuilderForm::class);
+  $entity_type->setFormClass("default", TemplateBuilderForm::class);
 
   /** @var \Drupal\Core\Config\Entity\ConfigEntityType $entity_type */
   $entity_type = $entity_types['entity_template_blueprint'];
-  $entity_type->setListBuilderClass(\Drupal\entity_template_ui\Entity\TemplateBlueprintListBuilder::class);
+  $entity_type->setListBuilderClass(TemplateBlueprintListBuilder::class);
   $entity_type->setHandlerClass("route_provider", [
-    "html" => \Drupal\Core\Entity\Routing\DefaultHtmlRouteProvider::class,
+    "html" => DefaultHtmlRouteProvider::class,
   ]);
   $entity_type->setLinkTemplate("collection", "/admin/content/entity_template/blueprint");
   $entity_type->setLinkTemplate("canonical", "/admin/content/entity_template/blueprint/manage/{entity_template_blueprint}");
   $entity_type->setLinkTemplate("edit-form", "/admin/content/entity_template/blueprint/manage/{entity_template_blueprint}/edit");
   $entity_type->setLinkTemplate("add-form", "/admin/content/entity_template/blueprint/add");
-  $entity_type->setFormClass("default", \Drupal\entity_template_ui\Form\TemplateBlueprintForm::class);
+  $entity_type->setFormClass("default", TemplateBlueprintForm::class);
 }
 
 /**
diff --git a/src/MultipleTemplateResult.php b/src/MultipleTemplateResult.php
index 719565c..19b800a 100644
--- a/src/MultipleTemplateResult.php
+++ b/src/MultipleTemplateResult.php
@@ -21,7 +21,7 @@ class MultipleTemplateResult extends TemplateResult {
   /**
    * {@inheritdoc}
    */
-  public function getEntity(): \Drupal\Core\Entity\EntityInterface {
+  public function getEntity(): EntityInterface {
     return reset($this->entities);
   }