diff --git a/examples/structured-outputs-image.py b/examples/structured-outputs-image.py
index 73d09cc..2de17b4 100644
--- a/examples/structured-outputs-image.py
+++ b/examples/structured-outputs-image.py
@@ -2,14 +2,13 @@
 from pydantic import BaseModel
 from typing import List, Optional, Literal
 from ollama import chat
-from rich import print
 
 
 # Define the schema for image objects
 class Object(BaseModel):
   name: str
   confidence: float
-  attributes: Optional[dict] = None
+  attributes: str
 
 
 class ImageDescription(BaseModel):