diff --git a/src/Extensions/MemberExtension.php b/src/Extensions/MemberExtension.php
index 69c02b2..0ae0235 100644
--- a/src/Extensions/MemberExtension.php
+++ b/src/Extensions/MemberExtension.php
@@ -3,7 +3,7 @@
 namespace SilverStripe\SessionManager\Extensions;
 
 use SilverStripe\Forms\FieldList;
-use SilverStripe\ORM\DataExtension;
+use SilverStripe\Core\Extension;
 use SilverStripe\ORM\HasManyList;
 use SilverStripe\Security\Member;
 use SilverStripe\SessionManager\FormFields\SessionManagerField;
@@ -14,9 +14,9 @@
  *
  * @method HasManyList<LoginSession> LoginSessions()
  *
- * @extends DataExtension<Member>
+ * @extends Extension<Member>
  */
-class MemberExtension extends DataExtension
+class MemberExtension extends Extension
 {
     /**
      * URL to the user help abot managing session
diff --git a/tests/php/Extensions/ForcePermission.php b/tests/php/Extensions/ForcePermission.php
index 01e6cdf..d0464c7 100644
--- a/tests/php/Extensions/ForcePermission.php
+++ b/tests/php/Extensions/ForcePermission.php
@@ -2,14 +2,14 @@
 
 namespace SilverStripe\SessionManager\Tests\Extensions;
 
-use SilverStripe\ORM\DataExtension;
+use SilverStripe\Core\Extension;
 use SilverStripe\Dev\TestOnly;
 
 /**
  * This extension is meant to be applied to LoginSession so we can force permission to scenarios that won't happen
  * natively with a default install of the session manager module.
  */
-class ForcePermission extends DataExtension implements TestOnly
+class ForcePermission extends Extension implements TestOnly
 {
     private static $canView = null;
     private static $canDelete = null;