diff --git a/modules/trusted_redirect_entity_edit/src/Service/EntityEditUrlResolver.php b/modules/trusted_redirect_entity_edit/src/Service/EntityEditUrlResolver.php
index a7ff0d0f2494b4b4775a575931e29912a8578095..b11bdcd92917b8e8d8e5cd5b0eeafde7401463e5 100644
--- a/modules/trusted_redirect_entity_edit/src/Service/EntityEditUrlResolver.php
+++ b/modules/trusted_redirect_entity_edit/src/Service/EntityEditUrlResolver.php
@@ -154,7 +154,8 @@ class EntityEditUrlResolver {
    *   Uuid of entity.
    *
    * @return bool|\Drupal\Core\Entity\EntityInterface
-   *   Entity or false if not possible to load entity by uuid.
+   *   - TRUE if the entity exists.
+   *   - The parent entity if found.
    */
   protected function loadEntityByUuid($entity_type, $uuid) {
     try {
diff --git a/src/EventSubscriber/TrustedRedirectSubscriber.php b/src/EventSubscriber/TrustedRedirectSubscriber.php
index 3431845e8bf02d23be49e704e2f89c4bf4cf8c50..c424556e0027a26f78088dcb114cf39f675167cb 100644
--- a/src/EventSubscriber/TrustedRedirectSubscriber.php
+++ b/src/EventSubscriber/TrustedRedirectSubscriber.php
@@ -6,11 +6,11 @@ use Drupal\Component\HttpFoundation\SecuredRedirectResponse;
 use Drupal\Component\Utility\UrlHelper;
 use Drupal\Core\Routing\TrustedRedirectResponse;
 use Drupal\trusted_redirect\TrustedRedirectHelpersTrait;
+use Symfony\Component\EventDispatcher\EventSubscriberInterface;
 use Symfony\Component\HttpFoundation\RedirectResponse;
 use Symfony\Component\HttpFoundation\Request;
 use Symfony\Component\HttpKernel\Event\ResponseEvent;
 use Symfony\Component\HttpKernel\KernelEvents;
-use Symfony\Component\EventDispatcher\EventSubscriberInterface;
 
 /**
  * Trusted redirect subscriber to redirect to trusted hosts.