diff --git a/tests/PartialsTest.php b/tests/PartialsTest.php
index d7c879bc..433e1203 100644
--- a/tests/PartialsTest.php
+++ b/tests/PartialsTest.php
@@ -7,6 +7,7 @@
 use Spatie\LaravelData\DataCollection;
 use Spatie\LaravelData\Lazy;
 use Spatie\LaravelData\Resolvers\RequestQueryStringPartialsResolver;
+use Spatie\LaravelData\Resource;
 use Spatie\LaravelData\Support\Partials\Partial;
 use Spatie\LaravelData\Support\Partials\PartialsCollection;
 use Spatie\LaravelData\Support\Partials\PartialType;
@@ -869,7 +870,7 @@ public function __construct(
 });
 
 it('can work with lazy laravel data collections', function () {
-    $dataClass = new class () extends Data {
+    $dataClass = new class () extends Resource {
         #[DataCollectionOf(SimpleData::class)]
         public Lazy|Collection $lazyCollection;