Fixes an bug in ImageModel