diff --git a/tests/Services/GetCollectionTest.php b/tests/Services/GetCollectionTest.php
index 22f1ceb..9a8693e 100644
--- a/tests/Services/GetCollectionTest.php
+++ b/tests/Services/GetCollectionTest.php
@@ -5,6 +5,7 @@
 use AstrobinWs\Exceptions\WsException;
 use AstrobinWs\Exceptions\WsResponseException;
 use AstrobinWs\Response\DTO\AstrobinError;
+use AstrobinWs\Response\DTO\AstrobinResponse;
 use AstrobinWs\Response\DTO\Collection\ListCollection;
 use AstrobinWs\Response\DTO\Collection\ListImages;
 use AstrobinWs\Response\DTO\Item\Collection;
@@ -149,7 +150,7 @@ public function testAddImagesInCollection(): void
         $collection = $reflectionMethod->invoke($this->astrobinWs, $collection); // $this->astrobinWs->getImagesFromResource($collection);
         $this->assertCount($nbItems, $collection->images);
         foreach ($collection->images as $image) {
-            $this->assertInstanceOf(Image::class, $image);
+            $this->assertInstanceOf(AstrobinResponse::class, $image);
         }
     }