diff --git a/src/Extension/FileTextExtractable.php b/src/Extension/FileTextExtractable.php
index 245899f..5e9b40e 100644
--- a/src/Extension/FileTextExtractable.php
+++ b/src/Extension/FileTextExtractable.php
@@ -3,7 +3,7 @@
 namespace SilverStripe\TextExtraction\Extension;
 
 use SilverStripe\Assets\File;
-use SilverStripe\ORM\DataExtension;
+use SilverStripe\Core\Extension;
 use SilverStripe\TextExtraction\Cache\FileTextCache;
 use SilverStripe\TextExtraction\Extractor\FileTextExtractor;
 
@@ -15,9 +15,9 @@
  *
  * @author mstephens
  *
- * @extends DataExtension<File>
+ * @extends Extension<File>
  */
-class FileTextExtractable extends DataExtension
+class FileTextExtractable extends Extension
 {
     /**
      * @var array