Minor change to fix the image issue