diff --git a/tests/Unit/Resources/ContentTest.php b/tests/Unit/Resources/ContentTest.php
index 9d65818..1850289 100644
--- a/tests/Unit/Resources/ContentTest.php
+++ b/tests/Unit/Resources/ContentTest.php
@@ -7,6 +7,7 @@
 use GeminiAPI\Enums\MimeType;
 use GeminiAPI\Enums\Role;
 use GeminiAPI\Resources\Content;
+use GeminiAPI\Resources\Parts\FilePart;
 use GeminiAPI\Resources\Parts\ImagePart;
 use GeminiAPI\Resources\Parts\TextPart;
 use PHPUnit\Framework\TestCase;
@@ -105,7 +106,7 @@ public function testFromArrayWithParts()
         ]);
         $parts = [
             new TextPart('this is a text'),
-            new ImagePart(MimeType::IMAGE_JPEG, 'this is an image'),
+            new FilePart(MimeType::IMAGE_JPEG, 'this is an image'),
         ];
         self::assertInstanceOf(Content::class, $content);
         self::assertEquals($parts, $content->parts);