Fixed the image types