diff --git a/src/MemberReportExtension.php b/src/MemberReportExtension.php
index d266f2d..92f0c01 100644
--- a/src/MemberReportExtension.php
+++ b/src/MemberReportExtension.php
@@ -1,7 +1,7 @@
 <?php
 namespace SilverStripe\SecurityReport;
 
-use SilverStripe\ORM\DataExtension;
+use SilverStripe\Core\Extension;
 use SilverStripe\ORM\FieldType\DBDatetime;
 use SilverStripe\Security\Group;
 use SilverStripe\Security\Permission;
@@ -13,9 +13,9 @@
  * Extends the {@see Member} class with additional descriptions for elements.
  * See {@see UserSecurityReport} for usage.
  *
- * @extends DataExtension<Member>
+ * @extends Extension<Member>
  */
-class MemberReportExtension extends DataExtension
+class MemberReportExtension extends Extension
 {
     /**
      * Set cast of additional fields
diff --git a/src/Subsites/SubsiteMemberReportExtension.php b/src/Subsites/SubsiteMemberReportExtension.php
index f491988..5285036 100644
--- a/src/Subsites/SubsiteMemberReportExtension.php
+++ b/src/Subsites/SubsiteMemberReportExtension.php
@@ -2,7 +2,7 @@
 
 namespace SilverStripe\SecurityReport\Subsites;
 
-use SilverStripe\ORM\DataExtension;
+use SilverStripe\Core\Extension;
 use SilverStripe\Security\Member;
 use SilverStripe\Subsites\Model\Subsite;
 
@@ -11,9 +11,9 @@
  *
  * @author Damian Mooyman <damian@silverstripe.com>
  *
- * @extends DataExtension<Member>
+ * @extends Extension<Member>
  */
-class SubsiteMemberReportExtension extends DataExtension
+class SubsiteMemberReportExtension extends Extension
 {
 
     /**