Fix image input issue