diff --git a/src/Extensions/ComposerUpdateExtension.php b/src/Extensions/ComposerUpdateExtension.php
index 010ca46..11ee087 100644
--- a/src/Extensions/ComposerUpdateExtension.php
+++ b/src/Extensions/ComposerUpdateExtension.php
@@ -2,14 +2,14 @@
 
 namespace BringYourOwnIdeas\UpdateChecker\Extensions;
 
-use SilverStripe\ORM\DataExtension;
+use SilverStripe\Core\Extension;
 
 /**
  * Describes any available updates to an installed Composer package
  *
  * Originally from https://github.com/XploreNet/silverstripe-composerupdates
  */
-class ComposerUpdateExtension extends DataExtension
+class ComposerUpdateExtension extends Extension
 {
     private static $db = [
         'VersionHash' => 'Varchar',