diff --git a/src/ErrorPageFileExtension.php b/src/ErrorPageFileExtension.php
index fc5e2ba..21a30a8 100644
--- a/src/ErrorPageFileExtension.php
+++ b/src/ErrorPageFileExtension.php
@@ -3,15 +3,15 @@
 namespace SilverStripe\ErrorPage;
 
 use SilverStripe\Assets\Shortcodes\FileShortcodeProvider;
-use SilverStripe\ORM\DataExtension;
+use SilverStripe\Core\Extension;
 use SilverStripe\ORM\DataObject;
 
 /**
  * Decorates {@see File} with ErrorPage support
  *
- * @extends DataExtension<FileShortcodeProvider>
+ * @extends Extension<FileShortcodeProvider>
  */
-class ErrorPageFileExtension extends DataExtension
+class ErrorPageFileExtension extends Extension
 {
     /**
      * Used by {@see File::handle_shortcode}