diff --git a/utils.py b/utils.py
index 7219b07..7890658 100644
--- a/utils.py
+++ b/utils.py
@@ -20,7 +20,7 @@ def temporary_env_var(key: str, new_value: str | None):
             del os.environ[key]
 
 
-def images_to_pillow(images: Tensor | list[Tensor]) -> list[Image]:
+def images_to_pillow(images: Tensor | list[Tensor]) -> list[Image.Image]:
     pillow_images = []
     for _bn, image in enumerate(images):
         i = 255.0 * image.cpu().numpy()