diff --git a/src/Extensions/UserFormUseQueuedExportExtension.php b/src/Extensions/UserFormUseQueuedExportExtension.php
index 80ef252..d217daa 100644
--- a/src/Extensions/UserFormUseQueuedExportExtension.php
+++ b/src/Extensions/UserFormUseQueuedExportExtension.php
@@ -6,9 +6,12 @@
 use SilverStripe\Forms\GridField\GridFieldExportButton;
 use SilverStripe\GridfieldQueuedExport\Forms\GridFieldQueuedExportButton;
 use SilverStripe\ORM\DataExtension;
+use SilverStripe\UserForms\Model\UserDefinedForm;
 
 /**
  * Adjusts UserDefinedForm to use GridFieldQueuedExportButton instead of GridFieldExportButton (the default)
+ *
+ * @extends DataExtension<UserDefinedForm>
  */
 class UserFormUseQueuedExportExtension extends DataExtension
 {