Minor patch on the model fix process.