Fix image generate