diff --git a/examples/image_augmentation.py b/examples/image_augmentation.py
index ea9fc86..1b73779 100644
--- a/examples/image_augmentation.py
+++ b/examples/image_augmentation.py
@@ -76,7 +76,7 @@ def _get_image():
 
 def _imshow(image: jnp.ndarray) -> None:
   """Showes the input image using PIL/Pillow backend."""
-  image = pil.fromarray(np.asarray(image * 255.).astype(np.uint8), "RGB")
+  image = pil.fromarray(np.asarray(image * 255.).astype(np.uint8), "RGB")  # pyrefly: ignore[bad-assignment]
   image.show()  # pyrefly: ignore[missing-attribute]