diff --git a/iframe_lazy_loading.module b/iframe_lazy_loading.module
index 9cb290d885e07bfdf161b27d1f13745304e44938..271106c3190702827bde567ff99be0f2299f5113 100644
--- a/iframe_lazy_loading.module
+++ b/iframe_lazy_loading.module
@@ -1,11 +1,19 @@
 <?php
 
 /**
- * @param \Drupal\Core\Template\Attribute $attributes
+ * @file
+ * Hooks for the iframe_lazy_loading module.
+ */
+
+use Drupal\Core\Template\Attribute;
+
+/**
+ * Add the loading="lazy" attribute.
  *
- * @return void
+ * @param \Drupal\Core\Template\Attribute $attributes
+ *   Attributes of the iframe element.
  */
-function _iframe_lazy_loading_add_loading_attribute(\Drupal\Core\Template\Attribute $attributes): void {
+function _iframe_lazy_loading_add_loading_attribute(Attribute $attributes): void {
   if (!$attributes->hasAttribute('loading')) {
     $attributes->setAttribute('loading', 'lazy');
   }
@@ -19,7 +27,7 @@ function _iframe_lazy_loading_add_loading_attribute(\Drupal\Core\Template\Attrib
  * @phpstan-param mixed[] $variables
  */
 function iframe_lazy_loading_preprocess_iframe(array $variables): void {
-  if ($variables['attributes'] instanceof \Drupal\Core\Template\Attribute) {
+  if ($variables['attributes'] instanceof Attribute) {
     _iframe_lazy_loading_add_loading_attribute($variables['attributes']);
   }
 }
@@ -32,7 +40,7 @@ function iframe_lazy_loading_preprocess_iframe(array $variables): void {
  * @phpstan-param mixed[] $variables
  */
 function iframe_lazy_loading_preprocess_video_embed_iframe(array $variables): void {
-  if ($variables['attributes'] instanceof \Drupal\Core\Template\Attribute) {
+  if ($variables['attributes'] instanceof Attribute) {
     _iframe_lazy_loading_add_loading_attribute($variables['attributes']);
   }
 }