Fix to allow models other than the pug.