diff --git a/autocomplete_flexible.api.php b/autocomplete_flexible.api.php
index 38ddb24..dc45df0 100644
--- a/autocomplete_flexible.api.php
+++ b/autocomplete_flexible.api.php
@@ -1,15 +1,14 @@
 <?php
 
+use Drupal\Core\Form\FormStateInterface;
 /**
  * @file
  * Autocomplete Flexible API documentation.
  */
-
 /**
  * @addtogroup hooks
  * @{
  */
-
 /**
  * Alter autocomplete flexible widget item label provided by other modules.
  *
@@ -23,6 +22,6 @@
  *     \Drupal\Core\Entity\EntityInterface object.
  *   - field_name: The name of the field being displayed.
  */
-function hook_autocomplete_flexible_widget_label(string &$label, \Drupal\Core\Form\FormStateInterface $form_state, array $context) {
+function hook_autocomplete_flexible_widget_label(string &$label, FormStateInterface $form_state, array $context) {
   $label = $context['entity']->label();
 }