Small update to image generation.