diff --git a/src/Extensions/UserFormUseQueuedExportExtension.php b/src/Extensions/UserFormUseQueuedExportExtension.php
index 3cce525..03ff5ef 100644
--- a/src/Extensions/UserFormUseQueuedExportExtension.php
+++ b/src/Extensions/UserFormUseQueuedExportExtension.php
@@ -5,15 +5,15 @@
 use SilverStripe\Forms\FieldList;
 use SilverStripe\Forms\GridField\GridFieldExportButton;
 use SilverStripe\GridfieldQueuedExport\Forms\GridFieldQueuedExportButton;
-use SilverStripe\ORM\DataExtension;
+use SilverStripe\Core\Extension;
 use SilverStripe\UserForms\Model\UserDefinedForm;
 
 /**
  * Adjusts UserDefinedForm to use GridFieldQueuedExportButton instead of GridFieldExportButton (the default)
  *
- * @extends DataExtension<UserDefinedForm>
+ * @extends Extension<UserDefinedForm>
  */
-class UserFormUseQueuedExportExtension extends DataExtension
+class UserFormUseQueuedExportExtension extends Extension
 {
     protected function updateCMSFields(FieldList $fields)
     {