diff --git a/src/ErrorPageControllerExtension.php b/src/ErrorPageControllerExtension.php
index cd95af8..d2ccc2f 100644
--- a/src/ErrorPageControllerExtension.php
+++ b/src/ErrorPageControllerExtension.php
@@ -9,13 +9,15 @@
 use SilverStripe\Control\HTTPRequest;
 use SilverStripe\Control\HTTPResponse_Exception;
 use SilverStripe\Core\Extension;
+use SilverStripe\Forms\Form;
 
 /**
  * Enhances error handling for a controller with ErrorPage generated output
+ *
+ * @extends Extension<Controller|Form>
  */
 class ErrorPageControllerExtension extends Extension
 {
-
     /**
      * Used by {@see RequestHandler::httpError}
      *
diff --git a/src/ErrorPageFileExtension.php b/src/ErrorPageFileExtension.php
index 0880eed..e30747a 100644
--- a/src/ErrorPageFileExtension.php
+++ b/src/ErrorPageFileExtension.php
@@ -2,15 +2,17 @@
 
 namespace SilverStripe\ErrorPage;
 
+use SilverStripe\Assets\Shortcodes\FileShortcodeProvider;
 use SilverStripe\ORM\DataExtension;
 use SilverStripe\ORM\DataObject;
 
 /**
  * Decorates {@see File} with ErrorPage support
+ *
+ * @extends DataExtension<FileShortcodeProvider>
  */
 class ErrorPageFileExtension extends DataExtension
 {
-
     /**
      * Used by {@see File::handle_shortcode}
      *