diff --git a/src/Element/AddressLookupElement.php b/src/Element/AddressLookupElement.php
index 6c86a58..d4d75d4 100644
--- a/src/Element/AddressLookupElement.php
+++ b/src/Element/AddressLookupElement.php
@@ -4,14 +4,14 @@
 
 use Drupal\Component\Utility\Html;
 use Drupal\Core\Form\FormStateInterface;
-use Drupal\Core\Render\Element\FormElement;
+use Drupal\Core\Render\Element\FormElementBase;
 
 /**
  * Provides a central hub address lookup element.
  *
  * @FormElement("localgov_forms_address_lookup")
  */
-class AddressLookupElement extends FormElement {
+class AddressLookupElement extends FormElementBase {
 
   /**
    * Static search string.