fixing the image generator?