Switch to using FormStateInterface.

In latest Drupal HEAD, $form_state is no longer an array, it's FormStateInterface.