diff --git a/inline_entity_form.api.php b/inline_entity_form.api.php
index ae3d5672a9363e4631440a266921b1a8d87c55e6..db7111bbb39de28ec63895e31d49e38a67092123 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');
 }