small bug fix in image model