diff --git a/src/RegistryImportFeed.php b/src/RegistryImportFeed.php
index 9c2e936..0fb0709 100644
--- a/src/RegistryImportFeed.php
+++ b/src/RegistryImportFeed.php
@@ -108,7 +108,7 @@ public function getStoragePath($modelClass = null)
     /**
      * Loop import files in the storage path and push them into an {@link ArrayList}
      *
-     * @return ArrayList
+     * @return ArrayList<RegistryImportFeedEntry>
      */
     public function getImportFiles()
     {
diff --git a/src/RegistryPageController.php b/src/RegistryPageController.php
index b498dd7..8f89cfc 100644
--- a/src/RegistryPageController.php
+++ b/src/RegistryPageController.php
@@ -18,6 +18,9 @@
 use SilverStripe\View\ArrayData;
 use SilverStripe\View\ViewableData;
 
+/**
+ * @extends PageController<RegistryPage>
+ */
 class RegistryPageController extends PageController
 {
     private static $allowed_actions = [
@@ -203,7 +206,7 @@ public function canSortBy($property)
      * Format a set of columns, used for headings and row data
      *
      * @param  int $id The result ID to reference
-     * @return ArrayList
+     * @return ArrayList<ArrayData>
      * @throws RegistryException If parameters are used in column names
      */
     public function Columns($id = null)