diff --git a/src/PageExtension.php b/src/PageExtension.php
index 5ccb128..836e54c 100644
--- a/src/PageExtension.php
+++ b/src/PageExtension.php
@@ -4,12 +4,12 @@
 
 use SilverStripe\Assets\File;
 use SilverStripe\Forms\FieldList;
-use SilverStripe\ORM\DataExtension;
+use SilverStripe\Core\Extension;
 
 /**
  * @method File ImportedFromFile()
  */
-class PageExtension extends DataExtension
+class PageExtension extends Extension
 {
 
     private static $has_one = [