diff --git a/src/Eloquent/Collection.php b/src/Eloquent/Collection.php
index db46961..b7e8f73 100644
--- a/src/Eloquent/Collection.php
+++ b/src/Eloquent/Collection.php
@@ -4,6 +4,12 @@
 
 use Illuminate\Database\Eloquent\Collection as Base;
 
+/**
+ * @template TKey of array-key
+ * @template TModel of \Illuminate\Database\Eloquent\Model
+ *
+ * @extends \Illuminate\Database\Eloquent\Collection<TKey, TModel>
+ */
 class Collection extends Base
 {
     public function toTree($childrenRelation = 'children')