Fix image model tests