Image generation fix.