diff --git a/inline_entity_form.api.php b/inline_entity_form.api.php
index 764aa0ad8008dd4730425a370f34201d6e77552b..3a9fcdc7717ad92e306c5555f66105f2821a6e55 100644
--- a/inline_entity_form.api.php
+++ b/inline_entity_form.api.php
@@ -10,10 +10,10 @@
  *
  * @param array $entity_form
  *   Nested array of form elements that comprise the entity form.
- * @param \Drupal\Core\Form\FormStateInterface $form_state
+ * @param $form_state
  *   The form state of the parent form.
  */
-function hook_inline_entity_form_entity_form_alter(array &$entity_form, FormStateInterface &$form_state) {
+function hook_inline_entity_form_entity_form_alter(array &$entity_form, \Drupal\Core\Form\FormStateInterface &$form_state) {
   if ($entity_form['#entity_type'] == 'commerce_line_item') {
     $entity_form['quantity']['#description'] = t('New quantity description.');
   }
@@ -27,10 +27,10 @@ function hook_inline_entity_form_entity_form_alter(array &$entity_form, FormStat
  *
  * @param array $reference_form
  *   Nested array of form elements that comprise the reference form.
- * @param \Drupal\Core\Form\FormStateInterface $form_state
+ * @param $form_state
  *   The form state of the parent form.
  */
-function hook_inline_entity_form_reference_form_alter(array &$reference_form, FormStateInterface &$form_state) {
+function hook_inline_entity_form_reference_form_alter(array &$reference_form, \Drupal\Core\Form\FormStateInterface &$form_state) {
   $reference_form['entity_id']['#description'] = t('New autocomplete description');
 }