diff --git a/resources/stubs/table.fillable.stub b/resources/stubs/table.fillable.stub
index 83ae916e..ec4a23b0 100644
--- a/resources/stubs/table.fillable.stub
+++ b/resources/stubs/table.fillable.stub
@@ -47,7 +47,7 @@ final class {{ componentName }} extends PowerGridComponent
     /**
     * PowerGrid datasource.
     *
-    * @return  \Illuminate\Database\Eloquent\Builder<\App\Models\User>|null
+    * @return  \Illuminate\Database\Eloquent\Builder<\{{ modelName }}>|null
     */
     public function datasource(): ?Builder
     {
diff --git a/resources/stubs/table.model.stub b/resources/stubs/table.model.stub
index 946df5af..0b751be0 100644
--- a/resources/stubs/table.model.stub
+++ b/resources/stubs/table.model.stub
@@ -47,7 +47,7 @@ final class {{ componentName }} extends PowerGridComponent
     /**
     * PowerGrid datasource.
     *
-    * @return  \Illuminate\Database\Eloquent\Builder<\App\Models\User>|null
+    * @return  \Illuminate\Database\Eloquent\Builder<\{{ modelName }}>|null
     */
     public function datasource(): ?Builder
     {