diff --git a/src/Extensions/ShareDraftContentSiteTreeExtension.php b/src/Extensions/ShareDraftContentSiteTreeExtension.php
index 7784619..edfb30c 100644
--- a/src/Extensions/ShareDraftContentSiteTreeExtension.php
+++ b/src/Extensions/ShareDraftContentSiteTreeExtension.php
@@ -15,7 +15,7 @@
 /**
  * @property SiteTree|ShareDraftContentSiteTreeExtension $owner
  * @property string $ShareTokenSalt
- * @method HasManyList|ShareToken[] ShareTokens()
+ * @method HasManyList<ShareToken> ShareTokens()
  */
 class ShareDraftContentSiteTreeExtension extends DataExtension
 {
diff --git a/src/Models/ShareToken.php b/src/Models/ShareToken.php
index 4e79b02..dd3f518 100644
--- a/src/Models/ShareToken.php
+++ b/src/Models/ShareToken.php
@@ -7,11 +7,10 @@
 use SilverStripe\ORM\FieldType\DBDatetime;
 
 /**
- * @method Page Page()
- *
  * @property int $ValidForDays
  * @property int $PageID
  * @property string $Token
+ * @method Page Page()
  */
 class ShareToken extends DataObject
 {