fixing the image issue