diff --git a/src/SiteTreeExtension.php b/src/SiteTreeExtension.php
index adb4ca2..a138d44 100644
--- a/src/SiteTreeExtension.php
+++ b/src/SiteTreeExtension.php
@@ -5,15 +5,15 @@
 use SilverStripe\CMS\Model\SiteTree;
 use SilverStripe\Forms\FieldList;
 use SilverStripe\Forms\Tab;
-use SilverStripe\ORM\DataExtension;
 use SilverStripe\Security\Permission;
+use SilverStripe\Core\Extension;
 
 /**
  * Handles adding the "Add to Campaign" button to a page's secondary actions menu
  *
- * @extends DataExtension<SiteTree>
+ * @extends Extension<SiteTree>
  */
-class SiteTreeExtension extends DataExtension
+class SiteTreeExtension extends Extension
 {
     protected function updateCMSActions(FieldList $actions)
     {