added in new model fix