diff --git a/stable_diffusion/txt2image.py b/stable_diffusion/txt2image.py
index a487cd22..877acae7 100644
--- a/stable_diffusion/txt2image.py
+++ b/stable_diffusion/txt2image.py
@@ -52,5 +52,5 @@ if __name__ == "__main__":
     x = (x * 255).astype(mx.uint8)
 
     # Save them to disc
-    im = Image.fromarray(x.__array__())
+    im = Image.fromarray(np.array(x))
     im.save(args.output)