Bugfix for image generation