diff --git a/src/Extensions/MaintenanceLockExtension.php b/src/Extensions/MaintenanceLockExtension.php
index 61063f1..7caf384 100644
--- a/src/Extensions/MaintenanceLockExtension.php
+++ b/src/Extensions/MaintenanceLockExtension.php
@@ -4,7 +4,7 @@
 
 use SilverStripe\Forms\CheckboxField;
 use SilverStripe\Forms\FieldList;
-use SilverStripe\ORM\DataExtension;
+use SilverStripe\Core\Extension;
 use SilverStripe\SiteConfig\SiteConfig;
 use Symbiote\QueuedJobs\Services\QueuedJobService;
 
@@ -14,9 +14,9 @@
  *
  * @package Symbiote\QueuedJobs\Extensions
  *
- * @extends DataExtension<SiteConfig&static>
+ * @extends Extension<SiteConfig&static>
  */
-class MaintenanceLockExtension extends DataExtension
+class MaintenanceLockExtension extends Extension
 {
     /**
      * @param FieldList $fields
diff --git a/src/Extensions/ScheduledExecutionExtension.php b/src/Extensions/ScheduledExecutionExtension.php
index 05f0c8b..a44874f 100644
--- a/src/Extensions/ScheduledExecutionExtension.php
+++ b/src/Extensions/ScheduledExecutionExtension.php
@@ -9,7 +9,7 @@
 use SilverStripe\Forms\NumericField;
 use SilverStripe\Forms\ReadonlyField;
 use SilverStripe\Forms\TextField;
-use SilverStripe\ORM\DataExtension;
+use SilverStripe\Core\Extension;
 use SilverStripe\ORM\DataObject;
 use SilverStripe\ORM\FieldType\DBDatetime;
 use Symbiote\QueuedJobs\DataObjects\QueuedJobDescriptor;
@@ -27,9 +27,9 @@
  *
  * @method QueuedJobDescriptor ScheduledJob()
  *
- * @extends DataExtension<DataObject&static>
+ * @extends Extension<DataObject&static>
  */
-class ScheduledExecutionExtension extends DataExtension
+class ScheduledExecutionExtension extends Extension
 {
     /**
      * @var array