Fix the image height code