Fixing the image import