diff --git a/src/Concerns/HasDrafts.php b/src/Concerns/HasDrafts.php
index e99f0c6..a12ff66 100644
--- a/src/Concerns/HasDrafts.php
+++ b/src/Concerns/HasDrafts.php
@@ -15,9 +15,9 @@
 use Oddvalue\LaravelDrafts\Facades\LaravelDrafts;
 
 /**
- * @method void Current(Builder $query)
- * @method void WithoutCurrent(Builder $query)
- * @method void ExcludeRevision(Builder $query, int | Model $exclude)
+ * @method Builder | Model current()
+ * @method Builder | Model withoutCurrent()
+ * @method Builder | Model excludeRevision(int | Model $exclude)
  */
 trait HasDrafts
 {