Fix failing model specs